Software Engineer, ML Inference Optimization
Foster City, CA
- Pay
USD 177,000–270,000/year — pay source
Salary 177,000 – 270,000 USD per year About Zoox
Read the full posting- Work setting
On-site — work setting source
Work setup in the employer’s posting: onsite
From the employer’s posting- Employment
Full-time — employment source
Employment type Full-time
Employment type in the employer’s posting: Full-time
From the employer’s posting
What you’ll work on
Full postingDesign, implement, and operate ML training or inference performance optimizations
Scale VLM, VLA, and foundation models for efficient deployment in robotaxis
Work with ML researchers, engineers and hardware teams on requirements and architecture
From the employer’s posting
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
What you’ll bring
All qualificationsCore experience
- 4+ years total experience, including 2+ years on large-scale training or inference platforms
- Experience with PyTorch and efficient distributed GPU training
- Experience with GPU-accelerated inference using TensorRT or similar
- Experience profiling with Nsight or PyTorch Profiler
- Proficient in Python or C++
Qualification wording
4+ years of total experience, including 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++
Tools in this posting
- C++
- Python
- PyTorch
Source — Tool mentions in context
- Experience using profiling tools like NVIDIA's Nsight or PyTorch's Profiler for identifying model training and serving bottlenecks. - Proficient in Python or C++. Base Salary Range
- 4+ years of total experience, including 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 with GPU-accelerated inference using TensorRT or similar frameworks. - Experience using profiling tools like NVIDIA's Nsight or PyTorch's Profiler for identifying model training and serving bottlenecks. - Proficient in Python or C++.
About Amazon
Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market.
In the employer’s words · Read in context
Posting record
Report an errorOpen 5 monthsWe first saw this posting 154 days ago and it is still up. Roles open this long are often slow to fill or kept open to collect applications.
- Status in our records
- Live
- First seen by us
- May 28, 2026 · 154 days ago
- Recorded sightings
- 206
- Last seen by us
- Oct 30, 2026
These dates show when we found the posting. Check the employer’s website to confirm it is still accepting applications.
Your next step
- Have your resume and examples of relevant work ready.
- Check the listed location, eligibility and core experience before starting.
Complete your application on jobs.lever.co. The employer’s form will show what is required.
Already applied? Track this application
Job description
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
- 4+ years of total experience, including 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 for identifying model training and serving bottlenecks.
- Proficient in Python or C++.
Base Salary Range
There are three major components to compensation for this position: salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights. A sign-on bonus may be offered as part of the compensation package. The listed range applies only to the base salary. Compensation will vary based on geographic location and level. Leveling, as well as positioning within a level, is determined by a range of factors, including, but not limited to, a candidate's relevant years of experience, domain knowledge, and interview performance. The salary range listed in this posting is representative of the range of levels Zoox is considering for this position.
Zoox also offers a comprehensive package of benefits, including 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, and life insurance.
Salary
177,000 – 270,000 USD per year
Employment type
Full-time
Source details
Source notes
Source excerptsSelected passages from the saved posting. Check the full description for conditions and exceptions.
- Pay
Salary 177,000 – 270,000 USD per year About Zoox
- Location & working pattern
Foster City, CA
Working pattern and location restrictions need checking in the full posting.
- Work authorization
No clear work-authorization passage found. Eligibility is unconfirmed.
See how this role fits your experience
Add your resume to compare the role’s scope, tools and requirements with your experience.