ML Performance Engineer — Distributed Training for Autonomy

jobs.frontdoordefense.com - Jobboard; Foster City (CA); On-site; USD 192,000 - 257,000; Full time

Benefits offered by this job

Health insurance

Paid time off

Stock options

Disability insurance

Job summary

Zoox is seeking a Software Engineer specializing in ML Performance Optimization to design and enhance our ML training and inference pipelines for autonomous driving. Located in Foster City, California, this role involves tackling cutting-edge challenges with SOTA accelerators.

You will collaborate across teams, drive performance optimization initiatives, and utilize your experience with frameworks like PyTorch. With a salary range of $192,000 to $257,000, Zoox offers a dynamic environment at the forefront of autonomous technology.

Qualifications

• 4+ years of total experience, including 2+ years of working on large-scale model training or inference platforms.
• Experience with training frameworks leveraging GPUs efficiently for distributed model training.
• Experience using profiling tools to identify model training and serving bottlenecks.

Responsibilities

• Design, implement, and operate performance optimization techniques for ML Training or Inference.
• Collaborate closely with cross-functional teams to align on requirements and architectural decisions.

Skills

Experience with large-scale model training

PyTorch proficiency

GPU-accelerated inference knowledge

Profiling tools experience

Proficient in Python

Proficient in C++

Job description

Zoox is seeking a Software Engineer specializing in ML Performance Optimization to design and enhance our ML training and inference pipelines for autonomous driving. Located in Foster City, California, this role involves tackling cutting-edge challenges with SOTA accelerators.

You will collaborate across teams, drive performance optimization initiatives, and utilize your experience with frameworks like PyTorch. With a salary range of $192,000 to $257,000, Zoox offers a dynamic environment at the forefront of autonomous technology.