# Software Engineer, ML Performance Optimization

**Company:** [Zoox](https://hotfix.jobs/companies/zoox)
**Location:** Foster City, CA
**Role:** ML Engineering
**Salary:** $192k – $257k/yr
**Experience:** 4+ years
**Skills:** PyTorch, TensorRT, Python, C++, Nvidia Nsight, Pytorch Profiler, Distributed Training, Quantization, Distillation, Pruning
**Posted:** 2026-05-28

> Drive ML performance optimization initiatives to make autonomous driving models faster and more efficient using distributed training, quantization, distillation, and profiling tools.

## Job Description

## Responsibilities
- Design, implement, and operate cutting-edge ML Training OR Inference performance optimization techniques to scale VLM, VLA, and Foundational models and deploy them efficiently in robotaxis.
- Collaborate closely with cross-functional teams, including ML researchers, software engineers, data engineers, and hardware engineers, to define requirements and align on architectural decisions.

## Requirements
- 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++.

## Nice-to-Haves
- Experience with distributed training techniques, quantization, distillation, and pruning.
- Work with SOTA accelerators and inference optimization frameworks.

## Similar roles

- [AI Engineer](https://hotfix.jobs/jobs/3e81f9d7-b85f-486a-849e-d1417b241b1e) - Zoox - Foster City, CA - $190k – $250k/yr
- [Software Engineer, Applied AI](https://hotfix.jobs/jobs/be4d97f1-9f2f-4973-bac2-0de1198d845e) - Norm Ai - New York, NY - $190k – $280k/yr
- [ML Engineer](https://hotfix.jobs/jobs/98ad2c38-046c-4b23-9f02-cb395af69959) - Sesame - San Francisco, CA - $190k – $320k/yr
- [Data Scientist, Underwriting](https://hotfix.jobs/jobs/4a054585-2634-407e-aa40-182679349330) - Parafin - San Francisco, CA - $190k – $300k/yr
- [AI Engineer](https://hotfix.jobs/jobs/3878e969-2b44-46f4-a8bf-9ceba03bdf47) - Eve - San Mateo, CA - $195k – $300k/yr

**Apply:** https://hotfix.jobs/jobs/b8ecd88e-b95d-4aa0-bf6c-4cee508dee44
**Canonical:** https://hotfix.jobs/jobs/b8ecd88e-b95d-4aa0-bf6c-4cee508dee44