Skip to content
NuroNuroMountain View, CA

Software Engineer, ML Infrastructure, Optimization

Build and optimize ML infrastructure for autonomous vehicles, focusing on model optimization, compilers, and deployment across the autonomy stack. Requires 2+ years in ML optimization and strong Python/C++/CUDA skills.

160k – 241k
On-site2+ YOEML Engineering

About the role

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.

Compensation and Benefits

  • Base pay range: $160,360 - $240,540
  • Annual performance bonus
  • Equity
  • Competitive benefits package

Skills

PythonC++CUDAPyTorchJAXTensorFlowKerasQuantizationModel PruningMl Compilers

Similar roles

ML Engineering jobs
Baseten

Software Engineer, Model Performance Tooling

BasetenSan Francisco, CA

Builds performance benchmarking, diagnostic, and optimization tools for LLM inference on GPU clusters. Early-career role requiring Python proficiency, systems curiosity, and interest in AI hardware—no prior experience needed.

160k – 200k
On-siteEntry levelML Engineering
Harvey

Software Engineer, Agents

HarveySan Francisco, CA +1

Builds AI agents for legal workflows, optimizing performance via prompt engineering, model selection, tools, and evaluations. Partners with PMs and customers to ship low-latency, high-quality agent systems using Python and LLM APIs. Requires 2-5 years experience.

161k – 242k
Hybrid2+ YOEML Engineering
Pinterest

Machine Learning Engineer, Monetization Engineering

PinterestSan Francisco, CA +2

Develops and evolves machine learning technology stack for Pinterest's Ads monetization, building personalized recommendation systems using deep learning and LLMs. Requires 2+ years ML experience, CS degree, and expertise in large-scale systems.

163k – 286k
Hybrid2+ YOEML Engineering
Abridge

Software Engineer

AbridgeSan Francisco, CA +1

Early-career Software Engineer building and iterating on agentic LLM systems, RAG pipelines, evaluation frameworks, and backend/frontend components for AI-powered clinical documentation at Abridge. Requires CS degree or equivalent plus hands-on GenAI experience from projects, internships or coursework.

157k – 184k
On-siteEntry levelML Engineering
Airbnb

Machine Learning Engineer, Relevance and Personalization

AirbnbUnited States

Build and productionize cutting-edge ML models and ranking algorithms for Airbnb's search, recommendation, and personalization systems at scale. Requires PhD (new grad) or 2+ years applied ML experience with strong programming and ML systems skills.

166k – 195k
Remote2+ YOEML Engineering