about 21 hours ago
Mountain View, CA, USA
Staff+
H1B Sponsor
Base Salary
$349k - $431k/yr
Responsibilities
- Define and drive the technical roadmap for the ML platform.
- Lead the unification of existing codebases into a production-hardened repository.
- Serve as the primary technical interface between Waymo and Google DeepMind.
- Drive the consolidation of core components across offboard Foundation Model use cases.
- Architect for efficient large-scale distributed training and knowledge distillation.
- Provide technical mentorship and guidance to engineers across multiple teams.
- Instill a culture of efficiency in model development and resource utilization.
Requirements
- Master's degree or PhD in Computer Science or a related field.
- 12+ years of experience in software engineering, with 8+ years in large-scale machine learning.
- A track record of architecting and delivering complex ML platforms.
- Deep expertise in Python, C++, and ML frameworks like JAX and TensorFlow.
- Extensive experience with large-scale distributed training on TPUs/GPUs.
- Ability to design robust, scalable, and maintainable software architectures.
- Understanding of data pipelines, storage systems, and tokenization techniques.
- Experience working effectively with research and product teams.
- Technical leadership skills to drive strategy and mentor engineers.
- Strong communication skills to articulate complex technical visions.
Benefits
- Eligible for Waymo’s discretionary annual bonus program.
- Participation in equity incentive plan.
- Generous Company benefits program.
Tech Stack
C++PythonTensorFlow
Categories
AI & MLBackendData Science