Position:
ML Engineer, Foundation Model Infrastructure
Company:
Waymo
Compensation:
$175,000—$215,000 USD
Location:
United States, Mountain View, California
Employment type:
Full-Time
Work Arrangement:
On Site
Short Summary:
Waymo is an autonomous driving technology company focused on building the Waymo Driver to improve access to mobility and save lives. The AI Foundations team develops machine learning solutions for autonomous driving, collaborating with other research teams in Alphabet.
Responsibilities:
- Build and operate petabyte-scale data systems and ML pipelines.
- Shepherd foundation models from research prototypes to robust components.
- Create automated infrastructure for benchmarking, monitoring, and releasing models.
- Utilize large-scale compute and frameworks like Flume and JAX.
- Partner with experts to transform model innovations into on-road improvements.
Requirement:
- Masters degree in Computer Science, Machine Learning, Robotics, or equivalent experience.
- Proficiency in Python and C++.
- Familiarity with modern deep learning frameworks (e.g., Pytorch, JAX, Tensorflow).
- Experience building or maintaining large-scale data pipelines or ML infrastructure.
Benefits:
- Top-notch medical, dental, and vision insurance.
- Competitive compensation, bonus opportunities, and equity.
- Flexibility to work from another location for four weeks per year.
- Enhanced leave options for birthing and non-birthing parents.
- Education reimbursement and personal development opportunities.
- Access to Google offices, wellness centers, and various perks.