Staff ML Engineer: Ultra-Fast Model Serving & Optimization

Waymo; Mountain View (NY); On-site; USD 251,000 - 310,000; Full time

Benefits offered by this job

Discretionary annual bonus program

Equity incentive plan

Generous company benefits

Job summary

Waymo is seeking a Machine Learning Engineer to optimize deep learning models and enhance simulation solutions. The ideal candidate will have a strong background in deep learning architectures and coding skills, particularly in Python.

Responsibilities include analyzing model performance, optimizing code for accelerators, and developing tools for analysis. Join us in pioneering technology that impacts mobility safety and innovation.

Qualifications

• 5+ years experience with deep learning architectures.
• Proficiency in JAX and Flax.
• Strong programming skills in Python.

Responsibilities

• Analyze model architectures and identify performance bottlenecks.
• Optimize model code for specific hardware accelerators.
• Design low-latency serving solutions for models.

Skills

Deep learning architectures

Optimization techniques

Programming skills in Python

Experience with JAX

Education

MS or PhD in Computer Science or related field

Tools

Profiling tools (e.g., XProf)

ML libraries (e.g., TensorFlow, PyTorch)

Job description

Waymo is seeking a Machine Learning Engineer to optimize deep learning models and enhance simulation solutions. The ideal candidate will have a strong background in deep learning architectures and coding skills, particularly in Python.

Responsibilities include analyzing model performance, optimizing code for accelerators, and developing tools for analysis. Join us in pioneering technology that impacts mobility safety and innovation.