Zoox
Autonomous Vehicles
SoftwareEngineer,MLPerformanceOptimization
Neural analysis suggests this role is
optimal for Mid candidates.
“Software Engineer, ML Performance Optimization at Zoox. Skills: ML Performance Optimization, Training frameworks, Inference optimization. Design ML Training OR Inference performance optimization techniques. Implement ML Training OR Inference performance optimization techniques”
What You'll Achieve.
Make Training and Inference platform fast and efficient
Industry & Context.
Optimize ML Training and Inference platform; Make platform fast and efficient
What They're Looking For.
Must Have
4+ years of total experience, 2+ years of working on large-scale model training or inference platforms, Experience with training frameworks like PyTorch, leveraging GPUs efficiently for distributed model training, Experience with GPU-accelerated inference using TensorRT or similar frameworks, Experience using profiling tools like NVIDIA's Nsight or PyTorch's Profiler, Proficient in Python or C++
What You'll Do.
Design ML Training OR Inference performance optimization techniques
Implement ML Training OR Inference performance optimization techniques
Operate ML Training OR Inference performance optimization techniques
and Foundational models
Deploy models efficiently
Build deep learning frameworks
Operate deep learning frameworks
Build inference systems
Operate inference systems
How You'll Work.
Team & Collaboration
Collaborate with cross-functional teams; Collaborate with ML researchers; Collaborate with software engineers; Collaborate with data engineers; Collaborate with hardware engineers; Define requirements; Align on architectural decisions; Act as a force multiplier for internal customers
Full Job Description
## Description Zoox is on a mission to reimagine transportation and ground-up build autonomous robotaxis that are safe, reliable, clean, and enjoyable for everyone. We are still in the early stages of deploying our robotaxis on public roads, and it is a great time to join Zoox and have a significant impact in executing this mission. The ML Platform team at Zoox plays a crucial role in enabling innovations in ML and CV to make autonomous driving as seamless as possible. The Opportunity Are you excited to lead our ML Performance Optimization initiatives and make our Training and Inference platform that enables autonomous driving as fast and efficient as possible? You will get to work across all ML teams within Zoox - Perception, Prediction, Planner, Simulation, Collision Avoidance, and Advanced Hardware Engineering group and have the opportunity to significantly push the boundaries of how ML is practiced within Zoox. We build and operate the base layer of ML tools, deep learning frameworks, and inference systems used by our applied research teams for in- and off-vehicle ML use cases. You will lead a team of strong software engineers and act as a force multiplier for our internal customers. This team has a lot of growth opportunities as we expand our robotaxi deployments and venture into new ML domains. If you want to learn more about our stack behind autonomous driving, please look here. ## In this role, you will Design, implement, and operate cutting-edge ML Training OR Inference performance optimization techniques to scale our VLM, VLA, and Foundational models and deploy them efficiently in our robotaxi. Collaborate closely with cross-functional teams, including ML researchers, software engineers, data engineers, and hardware engineers, to define requirements and align on architectural decisions. ## Qualifications Note: You do not have to meet all the requirements below to be considered for this position: 4+ years of total experience, including 2+ years of work
Applying for this Software Engineer, ML Performance Optimization role?
Most applicants get filtered before a human reads their resume. See if yours makes the cut.
How to Apply on Lever
- Lever uses a streamlined one-page form — apply in under 5 minutes.
- LinkedIn import works well; review parsed data before submitting.
- The cover letter field is optional but visible to reviewers — use it to differentiate.
- Referral codes from employees can significantly boost visibility of your application.
ANONYMOUS · UNFILTERED
What do employees actually say about Zoox?
Real rants from real employees. Read before you apply.