Senior ML Infra Engineer - Training Efficiency

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

Benefits offered by this job

Discretionary annual bonus program

Equity incentive plan

Generous Company benefits program

Job summary

A leading autonomous driving technology company in Mountain View is seeking an experienced professional to enhance ML infrastructure for training workloads. Responsibilities include designing distributed input data pipelines and collaborating with research teams to optimize systems. Candidates should have a strong background in Python and C++, as well as experience with ML frameworks like TensorFlow. This role offers competitive salary and benefits.

Qualifications

• B.S. in Computer Science, Math, or equivalent real-world experience.
• Solid programming skills in Python and C++.
• Hands-on experience with ML frameworks.

Responsibilities

• Design and improve distributed input data pipelines for ML workloads.
• Collaborate with researchers to resolve bottlenecks.
• Promote best practices for distributed systems architecture.

Skills

Distributed systems design

ML data pipeline optimization

Proficient in TensorFlow

Proficient in JAX

Python programming

C++ programming

Education

B.S. in Computer Science or Math

MS in Computer Science or Math (preferred)

Tools

ML Pathways

Grain

tf.data service

Job description

A leading autonomous driving technology company in Mountain View is seeking an experienced professional to enhance ML infrastructure for training workloads. Responsibilities include designing distributed input data pipelines and collaborating with research teams to optimize systems. Candidates should have a strong background in Python and C++, as well as experience with ML frameworks like TensorFlow. This role offers competitive salary and benefits.