Machine Learning Engineer, Model Optimization
Waymo
3 months ago
Mountain View, CA, USA or San Francisco, CA, USA
Mid Level / Senior
H1B Sponsor
Base Salary
$170k - $216k/yr
Responsibilities
- Optimize FLOPs utilization in model training and inference through architecture and hardware co-development.
- Enhance model inference for onboard and offboard simulation platforms.
- Analyze and optimize real-time inference of complex model architectures.
Requirements
- Bachelor's degree in Computer Science or a related field, or equivalent deep learning experience.
- 3+ years of experience in Machine Learning and/or Computer Vision.
- Proficiency in Python.
- Experience with ML frameworks such as PyTorch or JAX.
Benefits
- Eligible for Waymo's discretionary annual bonus program.
- Participation in equity incentive plan.
- Generous company benefits program.
Tech Stack
C++PythonPyTorch
Categories
AI & MLData Science