Current jobs related to Machine Learning Infrastructure Engineer - Mountain View, California - Nuro


  • Mountain View, California, United States Waymo Full time

    Job TitleSenior Machine Learning Engineer, TrainingAbout the RoleWe are seeking a Senior Machine Learning Engineer to join our Hybrid team at Waymo. As a key member of our ML Infrastructure team, you will be responsible for developing the infrastructure components necessary for distributed training, implementing automation solutions for provisioning,...


  • Mountain View, California, United States MatX Full time

    About the RoleThis Machine Learning Infrastructure Engineer position at MatX is an exciting opportunity to work on the development and deployment of next-generation AI systems. As a member of our team, you will be responsible for designing and building high-performance AI infrastructure, including distributed computing systems and large-scale data processing...


  • Mountain View, California, United States Nuro Full time

    **About Us**">Nuro is a robotics company that aims to improve everyday life through innovative technologies. Founded in 2016, we have spent years developing autonomous driving (AD) technology and commercializing AD applications. Our world-class autonomous driving system, the Nuro DriverTM, combines AD hardware with our AI-first self-driving software.We've...


  • Mountain View, California, United States NewsBreak Full time

    About UsAt NewsBreak, we are revolutionizing the way users interact with local news and their communities. Our mission is to foster safer, more vibrant, and authentically connected lives through robust collaborations with thousands of local publishers and businesses across the nation.We proudly stand as the nation's premier local news app, with our...


  • Mountain View, California, United States Nuro Full time

    Nuro is a pioneering robotics company dedicated to enhancing everyday life through innovative technology. Founded in 2016, we have spent years developing autonomous driving (AD) solutions and commercializing AD applications. Our world-class Nuro DriverTM combines AD hardware with AI-first self-driving software, built to learn and improve through data.We've...


  • Mountain View, California, United States NewsBreak Full time

    About NewsBreakWe're redefining the way users interact with local news and their communities. Our mission is to foster safer, more vibrant, and authentically connected lives by bridging local users, content creators, and businesses.As a Series-C unicorn startup, our headquarter is nestled in Mountain View, California, with other offices in New York City and...


  • Mountain View, California, United States Waymo Full time

    About the RoleWe're looking for a highly skilled Senior Machine Learning Engineer, Training to join our Waymo ML Infrastructure team. In this role, you'll develop infrastructure components for distributed training and implement automation solutions for provisioning, deployment, monitoring, and scaling of distributed training infrastructure.This Hybrid role...


  • Mountain View, California, United States NewsBreak Full time

    NewsBreak is revolutionizing the way users interact with local news and their communities. Our mission is to foster safer, more vibrant, and authentically connected lives.We are a Series-C unicorn startup based in Mountain View, California, with offices in New York City and Seattle.About NewsBreakWe are proud to be the nation's premier local news app. Since...


  • Mountain View, California, United States Waymo Full time

    About WaymoWaymo is an autonomous driving technology company dedicated to developing the world's most advanced driver.The Waymo Driver, our self-driving system, has been autonomously driving tens of millions of miles on public roads and simulating billions of miles in virtual environments across 13+ U.S. states.Job DescriptionWe are seeking a skilled...


  • Mountain View, California, United States Moveworks Full time

    Job DescriptionWe are seeking a highly skilled Senior Machine Learning Infrastructure Specialist to join our team at Moveworks. As a critical member of our AI infrastructure team, you will play a key role in building and optimizing cutting-edge machine learning systems for large language models.In this position, you will work closely with our...


  • Mountain View, California, United States NewsBreak Full time

    NewsBreak is redefining the way users interact with local news and their communities. Our mission is to foster safer, more vibrant, and authentically connected lives by bridging local users, content creators, and businesses.We are looking for a talented Machine Learning Infrastructure Developer to join our team. As a key member of our infrastructure team,...


  • Mountain View, California, United States Tik Tok Full time

    Job SummaryThe Machine Learning Infrastructure Specialist will be responsible for designing and implementing the infrastructure for TikTok's machine learning models. This role requires expertise in distributed systems, data engineering, and cloud computing.Key ResponsibilitiesDesign and develop scalable data pipelines for machine learning model training and...


  • Mountain View, California, United States Moveworks Full time

    At Moveworks, we're revolutionizing the way businesses interact with AI. As a Senior Machine Learning Infrastructure Specialist, you'll play a critical role in building and scaling our cutting-edge ML infrastructure.">We're looking for an expert in machine learning who can design, build, and optimize scalable ML infrastructure to support training,...


  • Mountain View, California, United States CV Library Full time

    Job OverviewWe are seeking a highly skilled Cloud Infrastructure Specialist to join our team at CV Library. This is a 12+ month contract opportunity that requires expertise in machine learning infrastructure, cloud platforms, and containerization technologies.Key Responsibilities:Design and implement scalable machine learning infrastructure on Google Cloud...


  • Mountain View, California, United States NewsBreak Full time

    Company Overview:NewsBreak is a pioneering local news app that has revolutionized the way users interact with their communities. Founded in 2015, the company has established itself as the nation's premier local news provider, bridging local users, content creators, and businesses across the nation.The company's headquarters is located in the tech hub of...


  • Mountain View, California, United States Nuro Full time

    About NuroNuro exists to better everyday life through robotics. Founded in 2016, we've spent eight years developing autonomous driving technology and commercializing applications.Our Driver is a world-class system combining AD hardware with our AI-first self-driving software. Built to learn and improve through data, it's one of the few driverless autonomous...


  • Mountain View, California, United States Waymo Full time

    **About Us**Waymo is a leading autonomous driving technology company dedicated to improving access to mobility while saving lives. Our mission is to be the most trusted driver, and we're committed to developing the world's most experienced driver - The Waymo DriverTM.We're seeking an exceptional Senior Machine Learning Engineer, Training to join our Hybrid...


  • Mountain View, California, United States Tik Tok Full time

    About the Role">TikTok is seeking a talented Machine Learning Engineer - Model Training Infrastructure to join our AML team. As a key member of this team, you will be responsible for designing and implementing a global-scale machine learning system for feeds, ads, and search ranking models.">Key Responsibilities">">Design and implement a global-scale machine...


  • Mountain View, California, United States Tik Tok Full time

    About Our Team We are a dynamic team of engineers working on building a scalable and secure machine learning infrastructure for TikTok. Our team is passionate about pushing the boundaries of AI innovation and is committed to delivering high-quality solutions that meet the needs of our users.Job Description We are seeking a talented Machine Learning Engineer...


  • Mountain View, California, United States Moveworks Full time

    About the RoleWe are seeking an experienced Machine Learning Engineer to help build cutting-edge ML infrastructure for large language models at Moveworks. This critical role will involve designing, building, and optimizing scalable machine learning systems for training, evaluation, and deployment of LLMs.The successful candidate will collaborate with our...

Machine Learning Infrastructure Engineer

1 month ago


Mountain View, California, United States Nuro Full time
About Nuro

Nuro exists to better everyday life through robotics. Founded in 2016, we have developed autonomous driving (AD) technology and commercialized AD applications. Our world-class autonomous driving system combines AD hardware with our generalized AI-first self-driving software. Our system is built to learn and improve through data and is one of the few driverless autonomous technologies on public roads today.

Nuro has raised over $2B in capital from leading investors. We've partnered with some of the world's most respected brands, including Uber, FedEx, Domino's, Walmart, Kroger, and 7-Eleven.

About the Role

We are growing our team and looking for talented engineers to join us. Nuro is pursuing a ML-first software stack. The ML Infra team is building scalable machine learning infrastructure and distributed training solutions to empower the ML development at Nuro. We are seeking an experienced engineer to join our team and contribute to this effort.

About the Work

Your primary responsibilities will include:
  1. Researching and developing new distributed training frameworks and strategies to support training deep learning models with growing sizes.
  2. Improving model training speed by optimizing TensorFlow, Keras, PyTorch, and CUDA kernel implementation.
  3. Engineering advanced tools to profile and monitor model training performance across all teams, to detect and triage training problems.

About You

We are looking for an experienced engineer with a strong background in machine learning and software development. You should have:
  1. 2+ years of relevant work experience, or an equivalent experience in PhD.
  2. Experience in building a cloud-based distributed training platform, which fully supports data and model parallelism.
  3. Experience investigating and optimizing training performance bottlenecks for deep learning models.

What We Offer

We offer a competitive salary range of $167,200 - $250,800 and an annual performance bonus. You will also receive equity and a competitive benefits package. We celebrate differences and are committed to a diverse workplace that fosters inclusion and psychological safety for all employees. Nuro is proud to be an equal opportunity employer and expressly prohibits any form of workplace discrimination based on race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, veteran status, or any other legally protected characteristics.