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.
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.
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
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
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
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.