about 2 hours ago
Mountain View, CA, USA
Senior
H1B Sponsor
Base Salary
$213k - $263k/yr
Responsibilities
- Architect and develop an efficient, high-performance ML runtime and serving system for autonomous vehicles and data centers.
- Lead integration and feature development for ML inference runtimes, balancing real-time latency and memory constraints.
- Drive the migration of ML workloads toward a JAX-native runtime architecture.
- Collaborate with Waymo ML practitioners to analyze workloads and apply hardware-aware optimizations.
- Design and build tooling for profiling, benchmarking, and identifying bottlenecks in the ML software stack.
Requirements
- B.S. or M.S. in CS, EE, Deep Learning, or a related field.
- 5+ years of professional software engineering experience in ML systems.
- 5+ years of production programming experience in C++.
- 3+ years of production experience in Python and major deep learning frameworks.
- Experience optimizing ML software for hardware accelerators.
- Experience building low-latency, highly concurrent distributed backend systems.
Benefits
- Eligible for Waymo’s discretionary annual bonus program.
- Participation in equity incentive plan.
- Generous Company benefits program.
Tech Stack
C++PythonPyTorch
Categories
AI & MLData Engineering