5 months ago
Foster City, CA, USAMid Level / Senior
H1B sponsor
Base Salary
$192k - $257k/yr
Responsibilities
- Design, implement, and operate ML performance optimization techniques for training and inference.
- Collaborate with ML researchers, software engineers, data engineers, and hardware engineers.
- Define requirements and align on architectural decisions across teams.
Requirements
- 4+ years of total experience, including 2+ years on large-scale model training or inference platforms.
- Experience with training frameworks like PyTorch and efficient GPU utilization.
- Experience with GPU-accelerated inference using TensorRT or similar frameworks.
- Familiarity with profiling tools like NVIDIA's Nsight or PyTorch's Profiler.
- Proficient in Python or C++.
About Zoox
Zoox builds a fully autonomous, purpose-built electric vehicle and the end-to-end platform to operate it as an urban robotaxi service. The company designs the vehicle, self-driving software, and fleet operations in-house to offer mobility-as-a-service rather than selling cars. Founded in 2014 and headquartered in Foster City, California, Zoox is a subsidiary of Amazon.