Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Software Engineer, ML Infrastructure, Optimization

$160.36k - $240.54k
Full-time

Nuro

Who We Are

Nuro believes self-driving vehicles are the most immediate and profound opportunity for AI to drive positive change in the physical world. Safer streets, more time for what matters, and easier access to the world around us, that’s why we’re building a universal autonomy platform: self-driving for all roads and all rides.

Founded in 2016, Nuro is a physical AI company developing Level 4 autonomous driving technology for a wide range of vehicles, use cases, and markets. Powered by the Nuro Driver™, our universal autonomy platform enables the global mobility ecosystem to deploy autonomy at scale, from robotaxis and logistics fleets to personal vehicles.

With years of real-world deployment experience and a flexible, partner-led business model, Nuro is working toward a future where millions of autonomous vehicles powered by our technology help make everyday life safer, easier, and more connected.

Nuro has raised over $2B in capital from Uber, NVIDIA, Google, Softbank, Fidelity, T. Rowe Price, and other leading investors.

About The Role

The Autonomy ML Infrastructure team is responsible for building & improving the core infrastructure for autonomy teams at Nuro. In this role, you will work closely with teams across Nuro, to design, build and deploy core infrastructure components in machine learning model life cycle, to push the autonomous future forward. You will have an opportunity to work across the full stack of machine learning solutions - from designing robust and scalable model pipelines to building to deploying the optimized models on Nuro’s fleet of self-driving robots!

About The Work

  • Optimize Nuro’s autonomy stack with cutting-edge optimization techniques like quantization, low precision inference, and model pruning.
  • Work with autonomy engineers to optimize, validate, and deploy large language models.
  • Develop and maintain a world-class model compiler framework, FTL.
  • Write robust, high-quality software to increase our confidence in our vehicle’s ability to navigate safely on-road.
  • Collaborate closely with machine learning domain experts and engineers across behavior, perception and mapping to design and implement end-to-end learned ML solutions.

About You

  • 2+ years of relevant experience in ML optimization infrastructure.
  • Experience with ML optimization techniques such as quantization and pruning, and ML compilers.
  • Experience maintaining, profiling, and optimizing GPU ML compilers & runtimes.
  • Proficient in Python and working experience with C++ and CUDA.
  • Working experience deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras).
  • Proficient in Python and working experience with C++.
  • You are passionate about accelerating the benefits of robotics for everyday life.

At Nuro, your base pay is one part of your total compensation package. For this position, the reasonably expected base pay range is between $160,360 and $240,540 for the level at which this job has been scoped. Your base pay will depend on several factors, including your experience, qualifications, education, location, and skills. In the event that you are considered for a different level, a higher or lower pay range would apply. This position is also eligible for an annual performance bonus, equity, and a competitive benefits package.

At Nuro, 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.
Vacancy posted 5 days ago
Similar jobs that could be interesting for youBased on the Software Engineer, ML Infrastructure, Optimization in California vacancy
  • $160.36k - $240.54k

     ...About the RoleThe Autonomy ML Infrastructure team is responsible for building...  ...building to deploying the optimized models on Nuro’s fleet of...  ...pruning.Work with autonomy engineers to optimize, validate, and...  ...Write robust, high-quality software to increase our confidence... 
    Suggested
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    1 day ago
  • $166k - $244k

     ...years of experience with software development in one or...  ...: Machine Learning Optimization (e.g., quantization, distillation...  ...in another ML field. 3 years of experience with ML infrastructure (e.g., model deployment...  ...Computer Science, Computer Engineering, or a related... 
    Suggested
    Full time
    Temporary work

    Google

    San Francisco, CA
    3 days ago
  • $175k - $220k

     ...building the future of generative AI infrastructure. Fireworks offers the generative...  ...As a Training Infrastructure Engineer, you'll design, build, and optimize the infrastructure that powers our...  ...experience with distributed systems and ML infrastructure ~ Experience with... 
    Suggested
    Full time

    Fireworks Ai

    Redwood City, CA
    1 day ago
  •  ...building the foundational infrastructure for the next...  ...development. We provide AI engineers and data scientists with...  ...The Role As a Software Engineer on our Infrastructure...  ...-based services. Optimize infrastructure for...  ...Experience with AI/ML workloads and infrastructure... 
    Suggested
    Full time
    Work at office
    Work from home
    1 day per week

    Runloop

    San Francisco, CA
    1 day ago
  • $200k - $300k

     ...Skild AI, Inc. seeks a Senior Software Engineer, AI Training & Infrastructure in San Mateo, CA. You will be responsible...  ...and tools that support the full ML lifecycle—data preparation,...  ...scale training, and evaluation. Optimizing training performance and resource... 
    Suggested
    Full time

    Skild AI

    San Mateo, CA
    1 day ago
  •  ...Specter is creating a software-defined "control plane" for...  ...ultimately become the perception engine for a company's physical...  ...Specter is hiring an ML infrastructure engineer to build and scale...  ...training, deployment, and optimization of computer vision and sensor... 
    Full time

    Specter

    San Francisco, CA
    1 day ago
  • $170k - $216k

     ...states. The Simulation Infrastructure team creates reliable,...  ...the Waymo Driver's software stack at a massive...  ...of customers Software Engineers, Product, Data Science...  ...Build and evolve ML inference infrastructure...  ...frameworks, TPUs and optimizing models for serving.... 
    Full time
    Remote work

    Waymo

    San Francisco, CA
    1 day ago
  •  ...functional group working across engineering, product, research, and...  ...for an experienced Software Engineer to help build the core infrastructure behind OpenAI’s...  ...delivery, measurement, and optimization Work closely with...  ...systems, marketplaces, AI/ML infra, or other... 
    Full time
    Relocation package
    Flexible hours

    OpenAI

    San Francisco, CA
    1 day ago
  • $196k - $339.9k

     ...work gets done. Airtable’s infrastructure is evolving to meet the needs of our fast growing engineering org. We are looking for...  ...support zero downtime failovers, optimizing performance and memory usage,...  ...and delivering cost efficient ML-ready datasets to power Airtable... 
    Remote job
    Full time
    For contractors
    Work at office
    Relocation
    Flexible hours

    Airtable

    San Francisco, CA
    1 day ago
  •  ...you choose to join us, will be as an Infrastructure / Backend Engineer on our founding team. You're the right...  ...in production Deploying and optimizing AI-heavy services with high availability...  ...deploying and productionizing ML or AI-heavy workloads Expertise in... 
    Full time

    Vooma

    San Francisco, CA
    1 day ago
  • $193.93k - $352.29k

     ...not fungible is the infrastructure that decides whether...  ...autonomously inside Nuro's own engineering organization, under...  ...About You3+ years of software engineering...  ...fine-tuning, preference optimization, and RL — and you can...  ...helped.Experience with ML training or research... 
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  •  ...a leader in AI cloud infrastructure serving tens of thousands...  ...Lambda Infrastructure Engineering organization forges...  ...an experienced Senior Software Engineer to join our storage...  ....Implement and optimize storage protocol APIs...  ...HaveExperience with AI/ML workloads and the unique... 
    Work at office
    Local area
    Work from home
    Flexible hours

    Lambda Labs

    San Jose, CA
    2 days ago
  • $293k - $385k

    About the TeamThe GPT Infrastructure team builds systems...  ...model inference and optimization into reliable production...  ..., performance engineering, secure partner integrations...  ...RoleWe are seeking a software engineer to help...  ...inference, distributed ML systems, or inference... 
    Work at office
    Local area
    Remote work
    Flexible hours

    OpenAI

    San Francisco, CA
    1 day ago
  • $180k - $300k

     ...to automatically curate and optimize petabytes of data to create...  ...both data research and data engineering necessary to solve this incredibly...  ...looking for an experienced Infrastructure Engineer to join as a member...  ...startup.Experience building ML/DL infrastructure and/or... 
    Work at office
    Work from home
    Relocation package

    DatologyAI

    Redwood City, CA
    3 days ago
  • $157k - $235k

     ...and other digital services.Snap Engineering teams build fun and technically sophisticated...  ...forefront.We’re looking for a Software Engineer to join the ML Platform Experience team, part of...  ...more.What you’ll do:Design and optimize infrastructure systems for machine learning... 
    Full time
    Live in
    Work at office
    Local area

    Snap

    Palo Alto, CA
    4 days ago
  • $174k - $252k

     ...one or more specialized ML areas, leverage ML infrastructure, and demonstrate expertise...  ...maintaining, or launching software products, and 1 year of...  ...deployment, model evaluation, optimization, data processing,...  ...technologies.Google's software engineers develop the next-... 

    Google

    San Bruno, CA
    4 days ago
  • $160.36k - $240.54k

     ...investorsAbout the RoleNuro is seeking a Software Engineer with expertise in large-scale infrastructure, workload orchestration, and data processing to join our ML Infrastructure team. In this role,...  ...Scheduling: Designing and optimizing workload orchestration to maximize... 
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    1 day ago
  • $147k - $211k

     ...in performance analysis, debugging, and optimization for distributed or low-level systems....  ...consensus algorithms (e.g., Paxos).Google's software engineers develop the next-generation...  ...loading pathways(essential for emerging AI/ML workloads), introducing advanced mutation... 

    Google

    Sunnyvale, CA
    1 day ago
  •  ...and builds the shared infrastructure that helps DoorDash, Wolt...  ...role is ideal for an engineer who enjoys building...  ....Partner closely with ML engineers, product engineers...  ...to agent optimization, where eval signals and...  ...industry experience in software engineeringStrong backend... 
    Hourly pay
    Work at office
    Local area
    Remote work
    Flexible hours

    Doordash

    San Francisco, CA
    4 days ago
  •  ...here. We are seeking a seasoned Software Engineer to build and scale the foundational compute infrastructure that powers our state-of-the-...  ...of every AI-generated video.Optimize GPU Utilization: Design and implement...  ...and Ray.Experience with core ML frameworks such as PyTorch,... 
    Full time

    HeyGen

    Los Angeles, CA
    3 days ago
  • $204k - $259k

     ...states. The Onboard Software Performance team...  ...experience as a software engineer or similar roles ~5...  ...~3 year of experience optimizing software using profile...  ...prefer: 2 years in infrastructure/systems/performance domain...  ...(SIMD/CUDA) and ML performance/frameworks... 
    Full time
    Work at office
    Remote work

    Waymo

    Mountain View, CA
    1 day ago
  •  ...More Big More Better). You will own optimizations on both the training and on-robot inference...  ...GPUs go brrrrr Implementing ML, hardware, and software changes that lead to step-function...  ...4 to hundreds of millions of users, engineered the foundations of autonomous driving... 
    Full time

    The Generalist

    San Francisco, CA
    1 day ago
  • $145k - $200k

     ...builds the world’s leading software for data-driven...  ...Role We are a software engineering team with expertise in enabling ML models in production. We...  ...at the intersection of infrastructure and machine learning that...  ...community enables us to optimize our opportunities to grow... 
    Full time
    Work experience placement
    Work at office
    Remote work
    Work from home
    Relocation package

    Palantir Technologies

    Palo Alto, CA
    4 days ago
  • $180k - $250k

     ...to automatically curate and optimize petabytes of data to create...  ...both data research and data engineering necessary to solve this incredibly...  ...for an experienced Cloud Infrastructure Engineer to join our core...  ...for training large-scale ML models Ensure our infrastructure... 
    Full time
    Work at office
    Relocation package

    Datologyai

    Redwood City, CA
    1 day ago
  • $192k - $257k

    Foster City, CASoftware - Software Systems /Full-time /On...  ...this mission. The ML Platform team at Zoox...  ...drive our ML Performance Optimization initiatives and make...  ...team of strong software engineers and act as a force...  ...learn more about our ML Infrastructure, here is one of our past... 
    Full time
    Temporary work
    Relocation package

    Zoox

    Foster, CA
    7 hours ago
  • $160k - $225k

     ...scale the foundational data infrastructure powering a category-...  ...product Work closely with engineering, data science, and product teams...  ...collaborate closely with data and ML teams and contribute to the...  ...and production use cases Optimize systems for scalability,... 
    Full time
    Work experience placement
    Relocation package
    Flexible hours

    Fable Security

    San Francisco, CA
    1 day ago
  • $160k - $225k

     ...scale the foundational data infrastructure powering a category-...  ...product Work closely with engineering, data science, and product teams...  ...reliability across pipelines Optimize systems for performance,...  ...engineers to productionize ML and analytics workflows... 
    Full time
    Work experience placement
    Relocation package
    Flexible hours

    Fable Security

    San Francisco, CA
    1 day ago
  •  ...and make it accessible to software developers of all skill levels...  ...scientist can scale an ML application from their...  ...is looking for a Software Engineer to join the Infrastructure team. Anyscale aims to provide...  ..., implementing, and optimizing the critical infrastructure... 
    Full time

    Anyscale

    San Francisco, CA
    1 day ago
  •  ...AI Engineer – Routing & Network Optimizatio About Gallatin At Gallatin, we are rebuilding...  ...for an AI Engineer – Routing & Network Optimization to own how supplies, vehicles, and...  ...-critical logistics. Experience with ML models, experimentation (A/B testing),... 
    Full time

    Gallatin

    El Segundo, CA
    1 day ago
  • $193.93k - $291.15k

     ...-output generalists where ML and systems engineering converge to push autonomy...  ...Senior Perception ML Data Infrastructure Engineer, you will own the...  ...High-Velocity Impact: You optimize the engineering lifecycle...  ...MS in Computer Science , Software Engineering, Robotics, or... 
    Full time

    Nuro

    Mountain View, CA
    1 day ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Software Engineer, ML Infrastructure, Optimization. Be the first to apply!