Skip to content
NuroNuroMountain View, CA

Senior/Staff Software Engineer, ML Infrastructure, Optimization

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
On-site3+ YOEML Engineering

About the role

About the Work

  • Optimize Nuro’s autonomy stack with cutting-edge optimization techniques like quantization, distillation, and model compression.
  • 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

  • 3+ 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.

Compensation and Benefits

  • Base pay range: $193,930 - $352,290
  • Annual performance bonus
  • Equity
  • Competitive benefits package

Skills

PythonC++CUDAPyTorchJAXTensorFlowKerasQuantizationPruningModel CompressionMl Compilers

Similar roles

ML Engineering jobs
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
On-site5+ 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
On-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
Hybrid7+ YOEML Engineering
Square

Staff Machine Learning Engineer, Credit Products (Square Financial Services)

SquareSeattle, WA +1

Develops and deploys ML models for credit underwriting in regulated banking, owning full stack from data curation to production decisioning. Leads MLOps initiatives and improves policies using advanced techniques for underserved segments. Requires 8+ YOE in ML production and technical degree.

195k – 343k
Hybrid8+ YOEML Engineering
Databricks

Staff Software Engineer

DatabricksNew York, NY

Staff ML Engineer building CustomerLake, Databricks' Customer Data Platform for enterprise ML/AI personalization, recommendations, churn, and LTV modeling. Requires 10+ years shipping production ML/LLM systems with strong product mindset in 0-to-1 environments.

192k – 260k
On-site10+ YOEML Engineering