Senior ML Architect - On-Device LLM/VLM Systems

Waymo; California (MO); On-site; USD 298,000 - 368,000; Full time

Benefits offered by this job

Annual bonus

Equity incentive

Benefits program

Job summary

Waymo is seeking an experienced ML engineer to advance large-scale model development (LLM/VLM) for on-device inference in autonomous driving contexts. You will collaborate with research, software, hardware, and product teams to push the boundaries of real-time, low-latency AI on embedded hardware.

Applicants should have a Master’s (or PhD) in a related field and a track record with PyTorch/JAX, foundation models, and safety-critical domains.

Qualifications

• 7+ years of ML experience with large-scale models.
• Experience with low-latency on-device inference.
• Proficient in PyTorch and JAX and large-scale model training.
• Background in LLMs or VLMs.
• Strong ability to operate under ambiguity.
• Experience applying foundation models in safety-critical domains.

Responsibilities

• Design and align model architectures with hardware constraints.
• Optimize on-device performance for real-time use cases.
• Collaborate with research, software, hardware, and product teams to deliver end-to-end solutions.

Skills

LLM/VLM

On-device inference

PyTorch/JAX

Ambiguity handling

Foundation models safety

Education

Master's degree

PhD (preferred)

Tools

PyTorch

JAX

Job description

Waymo is seeking an experienced ML engineer to advance large-scale model development (LLM/VLM) for on-device inference in autonomous driving contexts. You will collaborate with research, software, hardware, and product teams to push the boundaries of real-time, low-latency AI on embedded hardware.

Applicants should have a Master’s (or PhD) in a related field and a track record with PyTorch/JAX, foundation models, and safety-critical domains.