Skip to content
ZooxZooxFoster City, CA

Senior Machine Learning Engineer

Develop next-generation 3D occupancy and segmentation networks for autonomous vehicles by fusing Lidar, Camera, and Radar data into temporally consistent voxel representations. Requires MS/PhD + 6+ years experience in 3D CV, multi-modal fusion, and PyTorch.

242k – 290k/yr
Hybrid6+ YOEML Engineering

About the role

Responsibilities

  • Design and implement state-of-the-art multi-modal sensor fusion architectures (Lidar, Camera, Radar) to predict 3D occupancy, semantic segmentation, and flow.
  • Develop "vision-first" fusion strategies to enhance geometric understanding and reduce dependency on sparse sensor modalities.
  • Engineer temporal processing modules to improve the stability and consistency of predictions over time.
  • Optimize model architectures for real-time on-vehicle inference, balancing high-fidelity range extension with strict latency constraints.
  • Collaborate with downstream consumers (Tracking, Prediction, Planner) to refine geometric outputs, such as contours and free-space estimations, for complex maneuvering.

Requirements

  • MS or PhD in Computer Science, Robotics, Machine Learning, or related field.
  • 6+ years of industry experience.
  • Deep expertise in 3D Computer Vision and Deep Learning, specifically with voxel-based or BEV (Bird's Eye View) architectures.
  • Strong proficiency in Python and PyTorch for model training and design.
  • Some experience in C++ for model integration.
  • Experience with multi-sensor fusion (Lidar, Camera, Radar) and handling temporal data sequences.
  • Experience with occupancy networks, implicit representations (NeRF/Gaussian Splats), or scene flow estimation.

Nice-to-Haves

  • Experience optimizing models for TensorRT/CUDA to achieve low-latency inference.
  • Familiarity with sparse convolutions or query-based architectures for efficient 3D processing.
  • Experience with Vision Language Model, multi-modal 3D foundation model, World Model, or VLA.

Skills

3d computer visionDeep Learningvoxel-based architecturesbev architecturesPythonPyTorchC++multi-sensor fusionlidarcameraradartemporal dataoccupancy networksnerfgaussian splats

Similar roles

ML Engineering jobs
Zoox

Senior Machine Learning Engineer - ML Agents and Planning

ZooxFoster City, CA

Develops ML models using imitation and reinforcement learning for agent behavior prediction and autonomous vehicle planning. Requires PhD +1 year or MSc +5 years experience, expertise in RL, transformers, and production ML pipelines.

242k – 290k/yr
On-site1+ YOEML Engineering
Zoox

Senior Machine Learning Engineer - Perception Mapping

ZooxFoster City, CA

Design, train, and integrate ML models for semantic map element detection in autonomous vehicles. Requires 5+ years experience, MS/PhD in CS, expertise in computer vision, deep learning, and PyTorch.

242k – 290k/yr
On-site5+ YOEML Engineering
Zoox

Senior Machine Learning Engineer - Perception

ZooxFoster City, CA +1

Develops cutting-edge perception models using fused sensor data for autonomous vehicles, focusing on joint detection/tracking and embeddings for downstream systems. Requires MS/PhD in CS, PyTorch experience, and production deployment skills.

242k – 290k/yr
HybridML Engineering
Sigma

Senior AI/ML Engineer

SigmaSan Francisco, CA +1

Build and productionize AI/ML systems including recommendations, NLP interfaces, and agentic workflows. Requires 10+ years building production-grade AI/ML systems and experience with foundation models.

240k – 270k/yr
On-site10+ YOEML Engineering
Attentive

Senior Machine Learning Engineer

AttentiveSan Francisco, CA +1

Builds and operates scalable production ML systems for real-time personalization and targeting. Requires 6+ years experience with Python, PyTorch/TensorFlow, data pipelines, and cross-functional collaboration.

244k – 320k/yr
Hybrid6+ YOEML Engineering