Machine Learning Engineer (Infra), Driver Understanding and Evaluation

Added
1 day ago
Type
Full time
Salary
Upgrade to Premium to se...

Related skills

tensorflow pytorch simulation jax distributed training

📋 Description

  • Build scalable systems for training and fine-tuning large-scale models to evaluate driving
  • Work at the intersection of data engineering, model development, and simulation. Own large, complex
  • Contribute to production and optimization of ML models for Waymo’s fleet.
  • Design and scale distributed ML workflows supporting large datasets and model evaluation.
  • Collaborate cross-functionally to translate product goals into measurable technical deliverables.

🎯 Requirements

  • MS or PhD in CS/ML/AI or equivalent practical experience.
  • 3+ years in ML infrastructure: developing, scaling, training, deploying, and optimizing large-scale
  • Contributions to ML tooling/frameworks (e.g., PyTorch, Jax, TensorFlow, Ray); understand
  • Strong expertise in distributed training techniques and optimization for scaling across ML

🎁 Benefits

  • Salary range: $170,000 — $216,000 USD
  • Discretionary annual bonus, equity incentives, and benefits program
Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Engineering Jobs. Just set your preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.

Related Engineering Jobs

See more Engineering jobs →