Location: Mountain View (US-MTV-EMF690) San Francisco (US-SFO-MKT555), N/A, N/A, United States
Job Summary:
Job Duties:
- Lead model-platform co-optimization for Waymo's perception foundation model.
- Focus on training and inference efficiency on TPU, GPU, and CPU platforms.
- Collaborate with perception, simulation, and ML infrastructure teams.
Required Skills:
- Experience designing or optimizing neural networks (Transformers and ConvNets).
- Proficiency in optimizing model efficiency for large-scale training/inference.
- Ability to quickly learn new models and frameworks.
Required Experiences:
- Demonstrated track record of optimizing machine learning models.
- Experience with low-level optimizations on CPU/GPU/TPU preferred.
Job URLs: