ML Engineer, Foundation Model Infrastructure
Waymo
about 1 month ago
Mountain View, CA, USA or Remote, United States
Mid Level / Senior
H1B Sponsor
Base Salary
$204k - $259k/yr
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 and monitoring models.
- Utilize large-scale compute frameworks to process datasets and train models.
- Enhance the speed, reliability, and efficiency of the ML development lifecycle.
- Collaborate with AI Foundations and ML experts to implement model innovations.
Requirements
- Masters degree in Computer Science, Machine Learning, Robotics, or a related field.
- Proficiency in Python and C++.
- Familiarity with modern deep learning frameworks like Pytorch, JAX, or Tensorflow.
- Experience building or maintaining large-scale data pipelines or ML infrastructure.
Benefits
- Eligible for Waymo’s discretionary annual bonus program.
- Participation in equity incentive plan.
- Generous Company benefits program.
Tech Stack
Apache SparkC++PythonPyTorchTensorFlow
Categories
AI & MLData Engineering