Waymo is an autonomous driving technology company creating a new way forward in mobility.
Staff ML Engineer, Generative Model Performance & Efficiency
Location
United States
Posted
1 day ago
Salary
$251K - $310K / year
Seniority
Lead
Job Description
Staff ML Engineer, Generative Model Performance & Efficiency
Waymo
Role Description The Simulator Team at Waymo builds state-of-the-art simulations of realistic environments for testing, training, and validation of the Waymo Driver. Our team is a diverse, and collaborative group of machine learning (ML) engineers, software engineers, and ML research engineers. We develop industry-leading simulation solutions using advanced generative and reconstructive ML algorithms, to model the real world, encompassing realistic agents, roads, traffic systems, weather, and the full sensor suite (Camera, Lidar, Radar). To accelerate the fidelity, scalability, controllability, and richness of our simulations, we are pushing the frontiers of 3D world modeling. We leverage state-of-the-art ML technologies trained on large-scale datasets to create dynamic and semantically rich virtual worlds, directly impacting the development and validation of the Waymo Driver. In this role, you will report to a Senior Staff Engineering Manager. You will: - Analyze model architectures and identify bottlenecks in training and inference performance (e.g., memory bandwidth, compute, communication). - Apply and develop techniques such as quantization (e.g., FP8, INT4), pruning, knowledge distillation, and efficient attention mechanisms. - Optimize model code for specific hardware accelerators (TPUs, GPUs), leveraging compiler features and low-level libraries (e.g., XLA). - Experiment with different model partitioning and sharding strategies (e.g., data, tensor, pipeline parallelism, expert parallelism) to improve scalability and efficiency. - Design and implement low-latency, high-throughput serving solutions for generative models and optimize training pipelines to reduce training time. - Build and maintain tools for performance analysis, profiling (e.g., xprof), and debugging of ML models. Qualifications - MS or PhD in Computer Science, Machine Learning, Robotics, or a related field. - 5+ years of experience with deep learning architectures (especially Transformers, Diffusion Models, MoEs), algorithms, and optimization techniques. - Proficiency in JAX, Flax, and potentially TensorFlow/PyTorch. - Expertise in using profiling tools (e.g., XProf, Perfetto, NVIDIA Nsight) to diagnose performance issues in ML workloads. - Hands-on experience with quantization, pruning, distillation, and other model compression methods. - Strong programming skills in Python and potentially C++, with experience in software development best practices. Requirements - Knowledge of TPU and GPU architectures and how to optimize code for them. - Familiarity with ML compilers like XLA and an understanding of how they translate high-level code to efficient hardware instructions. - Understanding of concepts related to training and serving models across multiple devices and machines. - Experience contributing to frameworks and libraries that improve training speed and scalability (e.g., JAX, Gemax, XManager). Benefits - Waymo employees are eligible to participate in Waymo’s discretionary annual bonus program. - Equity incentive plan. - Generous Company benefits program, subject to eligibility requirements. Salary Range The expected base salary range for this full-time position across US locations is $251,000 — $310,000 USD. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level. Your recruiter can share more about the specific salary range for the role location or, if the role can be performed remote, the specific salary range for your preferred location, during the hiring process.
Related Guides
Related Categories
Related Job Pages
More Engineer Jobs
• Work collaboratively with our field-based FAE sales teams and account managers to maximize the number of opportunities entering the power sales funnel • Increase the possible number of opportunities to close across the UK • Design in of AC/DC, DC/DC and inverter customized power supplies • Supply value add solutions from our portfolio of power supply manufacturer partners and from our own power supply design center • Manage a sales project funnel pipeline from opportunities through to bookings • Propose optimal technical-commercial power solutions for our customers • Present and represent the Arrow Power Group • Deliver support and training for our sales representatives in technical issues • Grow the regions power supply bookings, billings, profit and design wins in line with targets
• Work directly (remotely or on-site when necessary) with customer engineering teams to design and implement real-time applications built on LiveKit • Design scalable architectures for voice AI, real-time media, and developer platform use cases • Build prototype integrations, reference implementations, and sample applications to accelerate customer development • Help customers navigate SDKs, APIs, infrastructure, and deployment strategies for production systems • Debug and troubleshoot complex technical issues in production environments • Lead technical workshops and architecture sessions with developers and product teams • Translate customer needs into technical insights that inform product and platform development • Partner with Product and Engineering to improve developer experience, documentation, and tooling • Contribute code, examples, or improvements to LiveKit's open-source ecosystem when helpful • Act as a trusted advisor helping customers successfully launch and scale applications on LiveKit
Structural Plan Review Engineer
SAFEbuiltSAFEbuilt provides community development services to more than 500 communities across the United States. The company serves local, county, and state governments of all sizes with q
Role Description We are currently seeking a talented and experienced Structural Plan Review Engineer to join our team of professionals. This position offers great opportunities for dynamic, motivated self-starters to work on a variety of work assignments. You’ll be a member of our highly respected team of professionals with a full complement of resources to support you and exciting projects to keep you challenged. - Review and evaluate structural construction documents submitted for approval utilizing applicable codes/standards/guidelines/laws, appropriate municipal ordinances, and construction and engineering standards, determine review fees, and authorize the release of approved documents for city permits. - Review structural engineering calculations for compliance with codes. - Provide timely oral and/or written communication detailing design and/or construction deficiencies in plans and specifications. - Direct and participate in design coordination meetings regarding private development projects and city, state, and regional capital improvement projects in order to lead the development process, ensure timely approval and completion of projects, and resolve any problem areas. - Assist inspectors with difficult or unusual code compliance issues and answer questions from inspectors about plan review comments. - Cultivate, foster, and maintain positive working relationships with directors, managers, supervisors, employees, and other stakeholders to gain their cooperation and support on assigned projects/assignments. Qualifications - Extensive knowledge of local building codes and current construction procedures/technology. - Interpretation of codes in the field and ability to provide code compliant solutions. - Must possess the ability to positively interact with the other staff members, the general public, contractors, governmental entities, and other clientele. - Excellent verbal and written communication skills. - Good time management, organization, and attention to detail skills. - Demonstrated ability to exercise initiative and a considerable amount of independent judgment. Requirements - Bachelor of Science Degree in Civil or Structural Engineering. - Registration as a Professional Engineer (P.E.) by a State Board of Registration. - ICC Building Plans Examiner Certification is a plus. - 3+ years of experience as a Licensed Professional Engineer working as a lead project engineer in the construction industry. - Experience performing wood-framed structural reviews. - Experience performing Commercial Building Plan Review is preferred. - Experience performing structural reviews for a municipality preferred. Benefits - SAFEbuilt offers a competitive salary and 401K with matching for part time employees.
Sr Elastic Engineer
ECS Tech IncAll candidates must meet the following criteria: Must be a US Citizen, no dual Citizenships. Must be able to secure a Public trust clearance. Must be able to work across multiple programs across the Federal and DOD space. The core values that ECS looks for in an engagement manager include: Teamwork, Respect, Accountability, Integrity, and Leadership.
Role Description ECS is seeking a Sr. Elastic Engineer to work remotely. This position is contingent upon additional funding. Are you passionate about building and scaling Elastic environments and eager to make an immediate technical impact? In this role, you will implement, optimize, and maintain Elastic Stack environments at scale while contributing to architecture, automation, and integrations that improve reliability, performance, and operational efficiency. We are seeking a Sr. Elastic Engineer to join our Professional team. The ideal candidate has hands-on experience with Elastic Stack and enjoys designing, coding, and deploying complex systems. You will collaborate with engineering, DevOps, cloud, and client teams to deliver resilient, high-performance Elastic solutions while maintaining visibility into system health and operational workflows. Key Responsibilities: - Design, deploy, and maintain Elastic Stack environments, including Elasticsearch, Kibana, Logstash, Beats, and Fleet. - Build and optimize ingestion pipelines, index strategies, search queries, dashboards, and automation workflows. - Write scripts, automation, and integrations (Python, PowerShell, Bash, etc.) to streamline operations, data processing, and monitoring. - Deploy Elastic across platforms, including on-premises, public cloud (AWS, Azure, GCP), GovCloud, and containerized environments (Kubernetes, Docker, OpenShift). - Leverage automation and orchestration tools such as Terraform, Ansible, CI/CD pipelines, and Docker to manage deployments and operational tasks. - Integrate Elastic with enterprise tools such as SIEMs, Splunk, CrowdStrike, and other telemetry sources. - Monitor system health, troubleshoot performance issues, and optimize for reliability and scalability. - Lead design reviews, provide guidance on Elastic best practices, and support knowledge sharing across teams. - Maintain documentation for architectures, deployment patterns, runbooks, and operational best practices. - Stay current with Elastic features, certifications, and emerging platform capabilities. Qualifications - Deep, hands-on expertise with Elastic Stack (Elasticsearch, Kibana, Logstash, Beats, Fleet). - Strong scripting and automation skills (Python, PowerShell, Bash, etc.). - Solid understanding of observability, logging, metrics, and distributed systems. - Experience designing, deploying, and optimizing production-scale Elastic environments. - Strong Linux/Unix, networking, and cloud platform knowledge (AWS, Azure, GCP). - Ability to mentor, guide, and influence engineering teams on Elastic best practices. - Outstanding verbal and written communication skills. - Willingness and ability to support domestic or international on-site engagements. - U.S. Passport required. - Must have a Secret Clearance, at minimum. Requirements - Salary Range: $170,000-$190,000 Benefits - General Description of Benefits
