Senior ML Infra Architect for Large-Scale Simulations

Waymo; Mountain View (CA); On-site; USD 204,000 - 259,000; Full time

Benefits offered by this job

Annual bonus program

Equity incentive plan

Generous Company benefits

Job summary

A leading autonomous technology firm in California seeks a Senior IC to advance AI/ML infrastructure for large models. This role involves collaborating with global teams to enhance simulation realism and designing distributed systems for ML lifecycles. Candidates should have over 5 years of software engineering experience, particularly in machine learning infrastructure, and familiarity with tools like PyTorch and TensorFlow is preferred. Strong leadership and communication skills are essential.

Qualifications

• 5+ years in software engineering with 3+ years in machine learning infrastructure.
• Experience with machine learning optimization and tools.
• Strong understanding of machine learning models and algorithms.

Responsibilities

• Lead the development of AI/ML infrastructure for large foundation models.
• Collaborate with teams to improve simulation realism.
• Design and scale distributed systems that support ML lifecycles.

Skills

Software engineering

Machine learning infrastructure

Large-scale distributed systems

Communication skills

Tools

DeepSpeed

PyTorch

TensorFlow

Ray

Job description

A leading autonomous technology firm in California seeks a Senior IC to advance AI/ML infrastructure for large models. This role involves collaborating with global teams to enhance simulation realism and designing distributed systems for ML lifecycles. Candidates should have over 5 years of software engineering experience, particularly in machine learning infrastructure, and familiarity with tools like PyTorch and TensorFlow is preferred. Strong leadership and communication skills are essential.