Build and operate scalable ML training frameworks, distributed infrastructure, and model lifecycle tools to power foundational models, reinforcement learning, and other ML use cases across Zoox's autonomous driving teams. Requires 2+ years ML infrastructure experience with PyTorch, DeepSpeed, JAX, Ray and AWS.
182k – 257k/yr
Hybrid2+ YOEML Engineering
About the role
Responsibilities
Build the Zoox Training framework leveraged by all ML teams within Zoox. This framework needs to be highly scalable, reliable, and efficient.
Design, implement, and operate a robust and efficient ML platform to enable the training, validation, serving, and monitoring of ML models.
Collaborate closely with cross-functional teams, including ML researchers, software engineers, and data engineers, to define requirements and align on architectural decisions.
Requirements
2+ years of ML infrastructure experience.
Experience with training frameworks like PyTorch, DeepSpeed, JAX, Ray, etc.
Experience working with cloud providers like AWS.
Nice-to-Haves
Experience with building large-scale, cost-efficient distributed model training and ML compute infrastructure.
Experience with building model lifecycle management tools and experimentation.
Designs datasets, evaluation rubrics, and reward signals for RLHF/RLVR pipelines to expose model failure modes and improve frontier AI capabilities. Partners with AI labs; requires 1-4 YOE and passion for data-driven model behavior.
180k – 220k/yr
On-site1+ YOEML Engineering
Forward-Deployed Engineer (CoE)
Tessera LabsNew York, NY
Leads end-to-end delivery of multi-agent AI systems for enterprise clients, analyzing workflows, building Python-based integrations with systems like SAP and Salesforce, and ensuring production deployment. Requires 1-3 years experience in data science or solutions engineering, Python proficiency, and strong communication skills.
180k – 233k/yr
Hybrid1+ YOEML Engineering
Forward Deployed ML Engineer
ModalNew York, NY +1
Partners with sales to drive technical sales of AI/ML infrastructure, leading demos, POCs, and solutions for enterprise customers. Requires 2+ years software engineering, AI/ML expertise, and strong communication skills.
180k – 250k/yr
On-site2+ YOEML Engineering
Machine Learning Engineer - 3D Sensor Simulation
ZooxFoster City, CA
Develops and optimizes ML-driven 3D sensor simulations (cameras, lidar, radar) using GenAI and graphics techniques to generate realistic synthetic data for AV testing. Requires 2+ years experience with neural rendering (NeRFs, Gaussian Splatting), PyTorch/TensorFlow, Python/C++, and 3D math.
176k – 257k/yr
Hybrid2+ YOEML Engineering
ML Engineer, Generative Video
MirageNew York, NY
ML Engineer building and scaling video generation models at Mirage. Focus on novel modeling, training optimization, inference efficiency (distillation, quantization, pruning), and low-latency real-time systems using PyTorch, CUDA, and distributed training. Requires 2+ years industry experience and strong deep learning systems expertise.