Current jobs related to Staff Machine Learning Engineer, Inference - Mountain View - Waymo


  • Mountain View, Arkansas, United States Waymo Full time

    Job Title: Senior Machine Learning Compiler EngineerWaymo is a leading autonomous driving technology company with a mission to create the most trusted driver. Our team is responsible for delivering the compute platform that powers our autonomous vehicles, and we're seeking a highly skilled Senior Machine Learning Compiler Engineer to join our ranks.About the...


  • Mountain View, California, United States IBM Full time

    About the RoleWe are seeking a highly skilled Machine Learning Engineer to join our team at IBM. As a key member of our conversational AI group, you will be responsible for designing, implementing, and deploying machine learning solutions to improve our conversational AI products.Key ResponsibilitiesInvestigate and experiment with new model architectures to...


  • Mountain View, Arkansas, United States Orby AI Full time

    Join Orby AI's Mission to Revolutionize AutomationOrby AI is building the next-generation Automation AI platform to automate repetitive processes. Our platform provides a unique "observe, learn, and automate" end-to-end experience powered by an actions-based foundation model. We observe a user's activities, identify repetitive work steps, and generate...


  • Mountain View, Arkansas, United States Waymo Full time

    Join Waymo's Compute TeamWaymo is a leader in autonomous driving technology, with a mission to improve access to mobility while saving thousands of lives. Our Compute Team is responsible for delivering the compute platform that powers our autonomous vehicles. We're seeking a talented Machine Learning Compiler Engineer to join our team.Key...


  • Mountain View, California, United States IBM Full time

    Job DescriptionWe are seeking a highly skilled Machine Learning Engineer to join our team at IBM. As a key member of our conversational AI group, you will be responsible for designing, developing, and deploying machine learning models to improve the accuracy and efficiency of our conversational AI system.Key Responsibilities:Investigate and experiment with...


  • Mountain View, California, United States Inworld AI Full time

    About Inworld AIInworld AI is a leading AI engine for games, enabling developers to build groundbreaking game mechanics, dynamic NPCs and worlds that evolve with each action. We power experiences built by top game developers and have partnerships with key industry players such as Microsoft/Xbox, Epic Games, and Unity.We are seeking a Staff/Principal Machine...


  • Mountain View, California, United States IBM Full time

    About the RoleWe are seeking a highly skilled Machine Learning Engineer to join our team at IBM. As a key member of our conversational AI group, you will play a critical role in designing, developing, and deploying advanced machine learning models that power our conversational AI system.Key ResponsibilitiesInvestigate and experiment with new model...


  • Mountain View, Arkansas, United States Waymo Full time

    Join Waymo's Compute TeamWaymo is a leader in autonomous driving technology, with a mission to create the most trusted driver. Our Compute Team is responsible for delivering the compute platform that powers our autonomous vehicles. We're seeking a talented Senior Machine Learning Compiler Engineer to join our team.About the RoleWe're looking for a skilled...


  • Mountain View, California, United States Moveworks Full time

    About the RoleMoveworks is seeking a highly skilled Machine Learning Engineer to join our team and help build cutting-edge ML infrastructure for large language models. This role will be critical in designing, building, and optimizing scalable machine learning systems to support training, evaluation, and deployment of LLMs.The successful candidate will work...


  • Mountain View, Arkansas, United States Waymo Full time

    Waymo is a pioneering autonomous driving technology company dedicated to making mobility safer and more accessible. Our mission is to be the most trusted driver, and we're committed to achieving this goal through innovative technology and collaboration.The Waymo Compute Team plays a critical role in this mission, responsible for delivering the compute...


  • Mountain View, California, United States Inworld AI Full time

    About Inworld AIInworld AI is a leading AI engine for games, enabling developers to build groundbreaking game mechanics, dynamic NPCs and worlds that evolve with each action.We are a well-funded startup with a $500 million valuation and backing from top-tier investors like Intel, Microsoft, Lightspeed, Bitkraft, Founders Fund, Kleiner Perkins, and more.We...


  • Mountain View, Arkansas, United States Waymo Full time

    Waymo Driver is a cutting-edge autonomous driving technology company with a mission to be the most trusted driver.Since its inception as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced DriverTM—to improve access to mobility while saving thousands of lives now lost to traffic...


  • Mountain View, Arkansas, United States Waymo Full time

    Waymo's Compute Team is tasked with a critical mission: delivering the compute platform responsible for running the autonomous vehicle's software stack. To achieve this, we architect and create high-performance custom silicon, develop system-level compute architectures, and collaborate with teammates to optimize hardware and software for maximum...

  • Software Engineer

    4 weeks ago


    Mountain View, California, United States Syntiant Full time

    About the RoleWe are seeking a highly skilled Software Engineer to join our team at Syntiant, a leading provider of AI solutions for embedded devices. As a Software Engineer on our team, you will be responsible for building and deploying machine learning solutions for our customers.As a key member of our team, you will work closely with our Core ML and...


  • Mountain View, Arkansas, United States Waymo Full time

    Job Title: Staff Machine Learning Compiler Engineer, ComputeWaymo is a leading autonomous driving technology company with a mission to create the most trusted driver. Our team is responsible for delivering the compute platform that powers our autonomous vehicles, and we're seeking a highly skilled Staff Machine Learning Compiler Engineer to join our Compute...


  • Mountain View, California, United States Inworld AI Full time

    About Inworld AIInworld AI is a leading AI engine for games, enabling developers to build groundbreaking game mechanics, dynamic NPCs and worlds that evolve with each action. Our technology powers experiences built by top game developers and has partnerships with key industry players such as Microsoft/Xbox, Epic Games, and Unity.We are seeking Staff and...

  • Software Engineer

    2 weeks ago


    Mountain View, California, United States Syntiant Full time

    Job OverviewSyntiant is seeking a highly skilled Software Engineer to join our Embedded and ML Frameworks team. As a key member of this team, you will be responsible for building machine learning solutions for embedded devices, working closely with our Core ML and Modeling teams to develop and deploy cutting-edge AI solutions.This role requires a strong...


  • Mountain View, Arkansas, United States Waymo Full time

    About the RoleWaymo is a leading autonomous driving technology company with a mission to create the most trusted driver. As a Staff Machine Learning Engineer, you will play a critical role in improving the performance of our car's machine learning models.Key ResponsibilitiesCollaborate with world-class Waymo ML scientists to identify opportunities for...


  • Mountain View, California, United States Gatik Full time

    About GatikGatik is a leader in autonomous middle mile logistics, delivering goods safely and efficiently using its fleet of light & medium-duty trucks. We focus on short-haul, B2B logistics for Fortune 500 customers, enabling them to optimize their hub-and-spoke supply chain operations, enhance service levels, and reduce labor costs.Job SummaryWe're seeking...


  • Mountain View, California, United States Coupang Full time

    Role OverviewCoupang's Search and Discovery Product Engineering (SDPE) ML group is a cutting-edge team that leverages machine learning to drive innovation and excellence in search and discovery experiences. As a Senior Staff Machine Learning Engineer, you will be part of this team, working on developing and deploying advanced machine learning models to...

Staff Machine Learning Engineer, Inference

2 months 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, debugging and evaluation, deployment, and monitoring. These efforts have resulted in making machine learning more accessible to teams at Waymo, ensuring greater degrees of consistency and repeatability. We work with machine learning experts in all parts of the company and our collaborators across Alphabet. You will focus on developing large-scale and efficient inference solutions for Waymo. This is crucial to the development of Waymo Foundation Models for driving including evaluation, auto-labeling and distillation.

In this hybrid role, you will report to the Senior Manager of Machine Learning.


You have:

  • B.S. in Computer Science, Math, or equivalent real-world experience
  • 2+ years of experience as a technical lead
  • 5+ years working with high-scale distributed or ML inference systems for real-time and batch processing
  • Understanding of machine learning fundamentals and experience with popular ML frameworks such as JAX, PyTorch, or TensorFlow

We prefer:

  • MS in Computer Science, Math, or equivalent real-world experience
  • Experience with multi-threaded and stream-based programming models
  • Prior experience on ML compiler optimization such as TVM or XLA
  • Previous experience deploying machine learning models for computer vision, natural language processing, or recommendation systems

#LI-Hybrid

#J-18808-Ljbffr