Zoox · 13 hours ago
Senior Machine Learning Engineer - 3D Segmentation
Boston, MA
Full-time
Hybrid
Senior Level
$242K/yr - $290K/yr
6+ years expZoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. In this role, you will contribute to the development of next-generation 3D occupancy and segmentation networks, optimizing high-performance deep learning models for vehicle navigation in complex urban scenarios.
Artificial Intelligence (AI)Autonomous VehiclesMachine LearningRoboticsTransportation
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
Qualification
Required
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
Preferred
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
Benefits
Paid time off (e.g. sick leave, vacation, bereavement)
Unpaid time off
Zoox Stock Appreciation Rights
Amazon RSUs
Health insurance
Long-term care insurance
Long-term and short-term disability insurance
Life insurance
Company
Zoox
Zoox is an AI robotics company that provides mobility-as-a-service and self-driving car services. It is a sub-organization of Amazon.
1001-5000 employees
H1B Sponsorship
Zoox has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (293)
2024 (297)
2023 (209)
2022 (204)
2021 (131)
2020 (83)
Funding
Current Stage
Late StageTotal Funding
$955MKey Investors
Grok VenturesBlackbird Ventures,Lux Capital,Thomas Tull,ThresholdLux Capital,Threshold
2020-06-26Acquired
2019-10-21Convertible Note· $200M
2018-07-08Series B· $465M
Leadership Team
Nelson Pedreiro
Sr. Vice President, Hardware
Z
Zheng Gao
Director of Hardware Engineering
Recent News
2026-01-16
Business Insider
2026-01-16
Channel NewsAsia
2025-12-24
Company data provided by crunchbase