Senior Software Engineer, ML Core
Zoox
about 1 month ago
Foster City, CA, USA
Senior
H1B Sponsor
Responsibilities
- Design, develop, and deploy custom and off-the-shelf ML libraries and tools.
- Build tooling and establish best practices for foundational libraries.
- Collaborate with cross-functional teams to define requirements and align on architectural decisions.
Requirements
- 6+ years of experience in software engineering.
- Proficient in Python or C++.
- Familiarity with training frameworks like PyTorch, Lightning, or JAX.
- Experience with GPU-accelerated inference on Nvidia hardware.
Tech Stack
BazelC++PythonPyTorch
Categories
AI & MLBackendData Science