Staff Machine Learning Engineer, Runtime

3 weeks ago


Mountain View, United States Waymo Full time

Waymo is an autonomous driving technology company with the mission to be the most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver-The World's Most Experienced Driver-to improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver powers Waymo One, a fully autonomous ride-hailing service, and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over one million rider-only trips, enabled by its experience autonomously driving tens of millions of miles on public roads and tens of billions in simulation across 13+ U.S. states.

The ML Platform team at Waymo provides a set of tools to support and automate the lifecycle of the machine learning workflow, including feature and experiment management, model development, optimization and monitoring. These efforts have resulted in making machine learning more accessible to teams at Waymo, including Perception, Planner, Research and Simulation.

We are looking for engineers with ML software or ML systems expertise to help us improve compute performance on both cloud and car. You'll work across the entire ML stack from the system perspective, from efficient deep learning models, model compression, ML software (e.g. JAX, XLA, Triton, and CUDA). You will be pleasantly challenged with deploying Waymo ML models on limited computation resources. In this hybrid role, you will report to the Senior Manager of Runtime and Optimization.

You will:
  • Lead the collaboration with the world-class Waymo ML scientists in perception, planner, research and simulation. Identify opportunities in both systems and models to make ML workloads faster.
  • Lead projects from proposals through execution by developing junior engineers.
  • Analyze and improve ML system workloads on both cloud and self-driving cars.
  • Apply model optimization, efficient deep learning techniques and ML software improvements to Waymo's ML systems.
You have:
  • M.S. in CS, EE, Deep Learning or a related field
  • 2+ years of experience as a technical lead, including writing project plans, engaging with customer teams, mentoring, responsible for goals & execution, reporting status.
  • 5+ years of experience developing solutions in ML systems or ML software stack (Pytorch/JAX/TF, runtime libraries, ML compiler).
  • Deep understanding of ML system architecture, performance analysis and tools.
  • Strong Python or C++ programming skills.
We prefer you have one or more of the following:
  • PhD in CS, EE, Deep Learning or a related field.
  • Familiarity with the HW architecture of ML hardware accelerators (e.g., GPU/TPU).
  • Deep knowledge of model optimization or efficient deep learning techniques for foundation models or LLM.
  • Experience with GPU HW or TPU HW and related system software.
#J-18808-Ljbffr

  • Mountain View, United States Intuit Full time

    Come join Intuit as a Staff Machine Learning Engineer!In this role, you’ll work alongside data scientists and machine learning engineers to create AI-powered experiences. You’ll be expected to help conceive, code, and deploy models at scale using the latest industry tools. Important skills include creating data pipelines, developing and deploying models,...


  • Mountain View, United States DeepRec.ai Full time

    Staff Machine Learning EngineerLocation: Onsite / Hybrid, Mountain View, CACompany: Stealth Mode GenAI StartupJoin a stealth mode GenAI startup backed by a top VC, they are currently a team of 10, with folks from Google, PayPal, and Coinbase to name a few. This company have recently secured a huge funding round and are focused on building cutting-edge...


  • mountain view, United States DeepRec.ai Full time

    Staff Machine Learning EngineerLocation: Onsite / Hybrid, Mountain View, CACompany: Stealth Mode GenAI StartupJoin a stealth mode GenAI startup backed by a top VC, they are currently a team of 10, with folks from Google, PayPal, and Coinbase to name a few. This company have recently secured a huge funding round and are focused on building cutting-edge...


  • Mountain View, United States Coupang Full time

    We exist to wow our customers. We know we’re doing the right thing when we hear our customers say, “How did we ever live without Coupang?” Born out of an obsession to make shopping, eating, and living easier than ever, we’re collectively disrupting the multi-billion-dollar e-commerce industry from the ground up. We are one of the fastest-growing...


  • Mountain View, Arkansas, United States Waymo Full time

    Job SummaryWaymo is an autonomous driving technology company with a mission to be the most trusted driver. The company has focused on building the Waymo Driver, the world's most experienced driver, to improve access to mobility while saving thousands of lives lost to traffic crashes. The Waymo Driver powers Waymo One, a fully autonomous ride-hailing service,...


  • Mountain View, United States Coupang Full time

    We exist to wow our customers. We know we’re doing the right thing when we hear our customers say, “How did we ever live without Coupang?” Born out of an obsession to make shopping, eating, and living easier than ever, we’re collectively disrupting the multi-billion-dollar e-commerce industry from the ground up. We are one of the fastest-growing...


  • Mountain View, United States Samsung Electronics Full time

    At Samsung Ads, we are actively distilling the maximum value out of data to drive business growth and solve complex problems. We are exploring the latest data mining and machine learning techniques to form a data-driven culture for decision-making and product design. As a data scientist of the Samsung Ads Platform Intelligence (PI) team, you will have access...


  • Mountain View, United States Gatik AI Full time

    Who we are Gatik, the leader in autonomous middle mile logistics, delivers goods safely and efficiently using its fleet of light & medium-duty trucks. The company focuses on short-haul, B2B logistics for Fortune 500 customers including Kroger, Walmart, Tyson Foods, Loblaw, Pitney Bowes, Georgia-Pacific, and KBX; enabling them to optimize their hub-and-spoke...


  • mountain view, United States WorldLink US Full time

    WHO we’re looking for:We are looking for a Machine Learning Engineer (Health) with solid ML/AI technology background and digital health-related research project experiences in health data analytics, physiological signal process, and multi-modality mobile sensing development. By leveraging smartphones, wearables, and hearables in the health/wellness domain,...


  • Mountain View, United States WorldLink US Full time

    WHO we’re looking for:We are looking for a Machine Learning Engineer (Health) with solid ML/AI technology background and digital health-related research project experiences in health data analytics, physiological signal process, and multi-modality mobile sensing development. By leveraging smartphones, wearables, and hearables in the health/wellness domain,...


  • Mountain View, United States Robert Half Full time

    Machine Learning Engineer, AI AdsGreat opportunity to join a thriving late-stage startup reshaping local news. They have over 40 million users and strong financial backing.This is a series-C unicorn media start-up empowering communities through innovative technology and partnerships.About the roleAre you a Machine Learning Engineer with a passion for...

  • Software Engineer

    1 month ago


    Mountain View, California, United States Databricks Full time

    About the RoleWe are seeking a highly skilled Software Engineer to join our Runtime Observability team at Databricks. As a key member of our engineering team, you will play a critical role in developing observability solutions that provide insights into the health and performance of our products and infrastructure.As a Software Engineer in this team, you...


  • Mountain View, United States META Full time

    Summary: Meta is embarking on the most transformative change to its business and technology in company history, and our Machine Learning Engineers are at the forefront of this evolution. By leading crucial projects and initiatives that have never been done before, you have an opportunity to help us advance the way people connect around the world. The ideal...


  • Mountain View, United States Samsung Electronics America North America Full time

    Samsung Ads is an advanced advertising technology company in rapid growth that focuses on enabling brands to connect with Samsung TV audiences as they are exposed to digital media by using the industry’s most comprehensive data to build the world’s smartest advertising platform. Being part of an international company such as Samsung and doing business...


  • Mountain View, United States Intuit Full time

    Overview In this role, you’ll be embedded inside a vibrant team of data scientists. You’ll be expected to help conceive, code, and deploy data science models at scale using the latest industry tools. Important skills include data wrangling, feature engineering, developing models, and testing metrics. You can expect to... What you'll bring Model...


  • Mountain View, United States Intuit Full time

    In this role, you’ll be embedded inside a vibrant team of data scientists. You’ll be expected to help conceive, code, and deploy data science models at scale using the latest industry tools. Important skills include data wrangling, feature engineering, developing models, and testing metrics. You can expect to...ResponsibilitiesDiscover data sources, get...


  • Mountain View, United States META Full time

    Summary: Meta is embarking on the most transformative change to its business and technology in company history, and our Machine Learning Engineers are at the forefront of this evolution. By leading crucial projects and initiatives that have never been done before, you have an opportunity to help us advance the way people connect around the world. The ideal...


  • Mountain View, United States Moveworks Full time

    ResponsibilitiesApply machine learning and software engineering to create lasting value for customersAddress challenging NLU, LLM, and conversational agent domain challengesEnsure responsible AI practices and privacy-preserving datasetsDesign new algorithms and architectures using machine learning fundamentalsResearch and develop innovative and scalable...


  • Mountain View, United States Intuit Full time

    Overview Intuit is looking for a highly motivated and experienced Principal Software Engineer to join the AI Synapse team. Our charter is to build AI/Machine Learning solutions for Intuit’s suite of financial products that drive quantifiable customer benefit through the state-of-the-art Large Language Models (LLMs) and Multimodal Language Models. You will...


  • Mountain View, United States Intuit Full time

    Intuit is looking for a highly motivated and experienced Principal Software Engineer to join the AI Synapse team. Our charter is to build AI/Machine Learning solutions for Intuit’s suite of financial products that drive quantifiable customer benefit through the state-of-the-art Large Language Models (LLMs) and Multimodal Language Models. You will be part...