Zoox Logo

Zoox

Senior Machine Learning Engineer - Perception 3D Segmentation

Posted 11 Days Ago
Be an Early Applicant
Hybrid
Foster City, CA, USA
242K-290K Annually
Senior level
Hybrid
Foster City, CA, USA
242K-290K Annually
Senior level
Develop and optimize multi-modal (Lidar, Camera, Radar) 3D occupancy, segmentation, and flow models. Architect temporal and vision-first fusion approaches, ensure temporal consistency, and optimize networks for real-time on-vehicle inference. Collaborate with tracking, prediction, and planning teams to deliver precise geometric and motion outputs for autonomous navigation.
The summary above was generated by AI
The Perception team at Zoox is responsible for the robot’s understanding of the world, fusing data from Lidar, Radar, and Cameras to create a unified representation of the environment. In this role, you will contribute to the development of our next-generation 3D occupancy and segmentation networks. You will architect and optimize high-performance deep learning models that generate dense, temporally consistent voxel representations of the driving environment. This work is critical for enabling our vehicle to navigate complex urban scenarios, handle rare obstacles, and drive safely in tight spaces by providing precise geometry and motion estimates to downstream planners.

In this role, you will...

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

Qualifications

  • MS or PhD in Computer Science, Robotics, Machine Learning, or related field with 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 deep learning frameworks (PyTorch) for model training and design as well as 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.

Bonus Qualifications

  • 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, or multi-modal 3D foundation model, or World Model, or VLA.

About Zoox
Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. Sitting at the intersection of robotics, machine learning, and design, Zoox aims to provide the next generation of mobility-as-a-service in urban environments. We’re looking for top talent that shares our passion and wants to be part of a fast-moving and highly execution-oriented team.

Follow us on LinkedIn

Accommodations
If you need an accommodation to participate in the application or interview process please reach out to [email protected] or your assigned recruiter.

A Final Note:
You do not need to match every listed expectation to apply for this position. Here at Zoox, we know that diverse perspectives foster the innovation we need to be successful, and we are committed to building a team that encompasses a variety of backgrounds, experiences, and skills.

HQ

Zoox Foster City, California, USA Office

4000 E 3rd Ave, Foster City, CA, United States, 94404

Zoox Foster City, California, USA Office

1149 Chess Drive, Foster City, CA, United States, 94404

Zoox Fremont, California, USA Office

47540 Kato Road, Fremont, CA, United States, 94538

Zoox San Francisco, California, USA Office

60 Broadway St, San Francisco, CA, United States, 94111

Similar Jobs

An Hour Ago
Easy Apply
Hybrid
Easy Apply
168K-224K Annually
Senior level
168K-224K Annually
Senior level
Cloud • Mobile • Software
Build and scale full‑stack product features and systems across frontend and backend. Own projects end‑to‑end from design through rollout and maintenance. Improve reliability, performance, observability, and scalability. Collaborate with Product and Design, mentor teammates, contribute to architecture and design reviews, and leverage AI tools to accelerate development while maintaining engineering judgment.
An Hour Ago
Easy Apply
Hybrid
Easy Apply
197K-262K Annually
Senior level
197K-262K Annually
Senior level
Cloud • Mobile • Software
Architect, build, and optimize a full-stack platform (React frontend, Node.js backend). Lead technical decisions, design and optimize APIs and databases, improve observability, enforce testing strategies, introduce new technologies, mentor engineers, and drive cross-team initiatives to deliver scalable, reliable software for commercial contractors.
Top Skills: ApolloAWSAzureCi/CdDockerGCPGraphQLJavaScriptJestKafkaKubernetesMaterial-UiMySQLNode.jsPostgresPythonRabbitMQReact Testing LibraryReactRelayRest ApisSQLTypescript
An Hour Ago
Easy Apply
Hybrid
San Francisco, CA, USA
Easy Apply
197K-262K Annually
Senior level
197K-262K Annually
Senior level
Cloud • Mobile • Software
Architect, build, and optimize a full-stack platform (React frontend, Node.js backends). Lead technical decisions, design scalable APIs, improve DB performance, implement observability, enforce testing, mentor engineers, and drive cross-team technical initiatives.
Top Skills: ApolloAWSAzureCi/CdDockerGCPGraphQLJavaScriptJestKafkaKubernetesMaterial-UiMySQLNode.jsPostgresPythonRabbitMQReactReact Testing LibraryReactRelayRest ApisSQLTypescript

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account