Software Engineer, ML Efficiency
Waymo
2 months ago
Mountain View, CA, USA +2 more
Entry Level / Mid Level
H1B Sponsor
Base Salary
$204k - $259k/yr
Responsibilities
- Enable the team to develop large scale end-to-end driving models with high hardware efficiency.
- Participate in model design from a hardware-efficiency perspective.
- Collaborate with other Waymo and Google teams on infrastructure work.
Requirements
- Bachelor degree in Computer Science or a similar technical field, or equivalent practical experience.
- Proficiency in writing and debugging Python/numpy-style code.
- In-depth knowledge of an ML framework such as Pytorch, JAX, or Tensorflow.
- Proficiency in system performance aspects like parallelism and asynchronicity.
Benefits
- Eligible for Waymo’s discretionary annual bonus program.
- Participation in equity incentive plan.
- Generous Company benefits program.
Tech Stack
C++NumPyPythonPyTorchTensorFlow
Categories
AI & MLData Science