Skip to content
NuroNuroMountain View, CA

Senior/Staff Engineer, Machine Learning - Online Mapping

Develops state-of-the-art online mapping models and algorithms for autonomous vehicles using real sensor data. Requires expertise in ML, robotics, computer vision, and deep learning frameworks like PyTorch.

184k – 334k/yr
On-siteML Engineering

About the role

About the Work

  • Research, develop, and implement state-of-the-art online mapping models and algorithms.
  • Analyze and characterize the performance of the online mapping system, identifying opportunities for architecture, data or evaluation improvements in a E2E ML system.
  • Work cross functionally with other ML teams to integrate our models into centralized architectures.
  • Collaborate with stakeholders across autonomy, infrastructure, and systems teams on online mapping needs and requirements.

About You

  • Proven record of solving in-production ML problems and making tradeoffs between data, model and evaluation.
  • Deep understanding of ML fundamentals with hands-on experience in training and evaluating modern ML models with applications in AV, robotics, mapping, computer vision, or related areas.
  • Prioritize impact and practicality, and make decisions to ensure on-time delivery of solutions.
  • Able to quickly iterate and experiment to cut through the noise and make well-informed, data-driven decisions.
  • Experience with robotics-related ML applications, 3D geometry.
  • Strong Python skills with experience in deep learning frameworks, e.g., PyTorch, TensorFlow, or Jax.

Bonus Points

  • Proficiency in working with complex multi-component systems.
  • Experience in building ML pipelines and optimizing/productizing ML models.
  • Familiarity with modern ML tools and infrastructure such as distributed training and ML compilers.
  • Demonstrated research publications in top conferences (e.g. NeurIPS, ICLR, ICML, CVPR, RSS, CoRL, ICRA).
  • Deep understanding of 3D geometry and state estimation fundamentals.

Compensation

Base pay range: $183,825 - $333,925. Eligible for annual performance bonus, equity, and competitive benefits package.

Skills

PyTorchTensorFlowJAXPythonMachine LearningComputer Vision3D GeometryLidarRoboticsMapping

Similar roles

ML Engineering jobs
Harper

Senior Member of Technical Staff, Harness Engineering

HarperSan Francisco, CA

As a Senior Member of Technical Staff, Harness Engineering, you will build the core substrate for Harper's AI agents, focusing on agent loop primitives, execution environment infrastructure, and tool layer ownership. You will work with frontier coding agents and internal agents, shipping primitives that agent teams depend on.

187k – 264k/yr
On-site3+ YOEML Engineering
xAI

Member of Technical Staff

xAIPalo Alto, CA

Build and optimize the RL training framework and infrastructure for large-scale workloads at SpaceXAI, from ablations to production runs. Requires experience with distributed systems and proficiency in Python, JAX, Rust, or C++.

180k – 440k/yr
On-site5+ YOEML Engineering
LlamaIndex

Member of Technical Staff, Applied Research

LlamaIndexSan Francisco, CA

Build and productionize vision-language models for document understanding at LlamaIndex. Focus on training, fine-tuning, synthetic data, benchmarking, and turning research prototypes into accurate, low-latency production systems for real-world PDFs, tables, and enterprise docs. Requires 3+ years ML engineering/applied research experience with strong PyTorch skills.

180k – 250k/yr
Hybrid3+ YOEML Engineering
xAI

Member of Technical Staff, Inference

xAIPalo Alto, CA

Build and optimize the high-performance inference platform serving Grok at massive scale. Design distributed serving infrastructure, low-level GPU optimizations, quantization, speculative decoding, and CI/CD for production reliability and low latency.

180k – 440k/yr
On-site5+ YOEML Engineering
Black Forest Labs

Member of Technical Staff

Black Forest LabsSan Francisco, CA

Research Engineer embedded in large-scale multimodal model training at Black Forest Labs (creators of FLUX). Own performance, stability, and low-precision optimizations for production training runs; profile, debug distributed systems, implement GPU kernels, and partner with researchers to ship frontier generative models.

180k – 290k/yr
Hybrid7+ YOEML Engineering