Skip to content
NuroNuroMountain View, CA

Senior/Staff Software Engineer, ML Inference Platform

Build and operate ML infrastructure for autonomy teams, including training and deployment pipelines, model observability, inference serving, and compiler platforms across hardware targets. Requires a degree, 3+ years of relevant experience, Python proficiency, and distributed-systems expertise.

194k – 352k/yr
On-site5+ YOEML Engineering

About the role

Responsibilities

  • Design and develop ML workflow pipelines to train, optimize, validate, and deploy autonomy models.
  • Develop and maintain continuous testing and monitoring systems for core ML infrastructure components.
  • Develop observability to track ML model lifecycles from data generation to on-road validation.
  • Maintain an in-house ML inference platform to serve large language models efficiently.
  • Maintain an in-house ML compiler platform to compile, deploy, and validate autonomy models across various hardware platforms.

Requirements

  • Bachelor’s, master’s, or Ph.D. degree.
  • 3+ years of relevant experience.
  • Strong proficiency in Python or similar languages.
  • Familiarity with C++.
  • Experience working with ML pipelines.
  • Ability to understand and design complex distributed systems.
  • Strong focus on performance, scalability, and code quality.
  • Ability to work independently, learn quickly, and improve complex systems.

Nice-to-Haves

  • Strong proficiency in C++ or other high-performance, low-level languages.
  • Experience working with large-scale distributed systems.
  • Experience with system and framework design.
  • Experience with data workflow orchestration platforms.
  • Experience with ML compilers.

Compensation and Benefits

  • Base pay range: $193,930–$352,290.
  • Eligible for an annual performance bonus, equity, and a competitive benefits package.

Skills

PythonC++machine learning pipelinesDistributed Systemsml inferenceml compilersWorkflow Orchestrationcontinuous testingObservabilityLLMs

Similar roles

ML Engineering jobs
Nuro

Senior/Staff Software Engineer, AI Agent Infrastructure

NuroMountain View, CA

Build and scale trustworthy AI agent infrastructure at Nuro, focusing on closed-loop evaluation, agent runtime/platform with isolation, and autoresearch systems for autonomous AI-driven engineering workflows. Requires deep LLM expertise, production agent experience, and strong backend/distributed systems skills.

194k – 352k/yrOn-site5+ YOEML Engineering
Nuro

Senior/Staff Machine Learning Engineer, Sensor Simulation

NuroMountain View, CA

Develop synthetic sensor simulation models and algorithms using ML techniques like NeRF and Gaussian splatting to generate photorealistic images and realistic lidar/radar data for autonomous vehicles. Requires advanced degree plus 3-5+ years experience, strong ML fundamentals, and Python/deep learning expertise.

194k – 291k/yrOn-site5+ YOEML Engineering
Nuro

Senior/Staff Software Engineer, ML Infrastructure, Optimization

NuroMountain View, CA

Build and optimize ML infrastructure for autonomous vehicles, focusing on model optimization, compilers, and deployment of large models to Nuro's self-driving fleet. Requires 3+ years ML optimization experience and strong Python/C++/CUDA skills.

194k – 352k/yrOn-site3+ YOEML Engineering
Nuro

Senior/Staff Software Engineer, ML Data

NuroMountain View, CA

Senior/Staff Software Engineer architects ML data pipelines for autonomy AI at Nuro, transforming massive datasets into training signals. Requires 7+ years experience, C++/Python proficiency, and expertise in end-to-end ML data systems.

194k – 352k/yrOn-site7+ YOEML Engineering
Okta

Staff Software Engineer, AI-Core

OktaSan Francisco, CA

Build and own core layers of a scalable Agentic AI platform for autonomous engineering agents at Okta, including agent identity/auth, knowledge/memory, observability, governance/safety, and orchestration. Requires 5+ years backend engineering experience plus AI/agent exposure; strong distributed systems and security knowledge.

194k – 267k/yrHybrid7+ YOEML Engineering