ML Performance Engineer for Scalable Foundational Models

Zoox; Foster City (CA); On-site; USD 192,000 - 257,000; Full time

Benefits offered by this job

Health insurance

Paid time off

Zoox Stock Appreciation Rights

Job summary

Zoox is seeking an ML Performance Optimization expert to improve the efficiency of their machine learning models for autonomous driving. This role involves working with cutting-edge technologies and collaborating with diverse teams to push the boundaries of machine learning.

The ideal candidate will have over 4 years of experience, significant knowledge of large-scale model platforms, and proficiency in Python or C++. A competitive salary with comprehensive benefits is offered, including RSUs and a sign-on bonus.

Qualifications

• 4+ years of total experience, including 2+ years in large-scale model training or inference.
• Experience with distributed model training using GPUs.
• Proficiency with GPU-accelerated inference frameworks.

Responsibilities

• Design and implement ML training performance optimization techniques.
• Collaborate with cross-functional teams on model deployment.

Skills

Large-scale model training platforms

GPU efficiency with training frameworks

Profiling tools like NVIDIA Nsight

Proficient in Python or C++

Tools

PyTorch

TensorRT

Job description

Zoox is seeking an ML Performance Optimization expert to improve the efficiency of their machine learning models for autonomous driving. This role involves working with cutting-edge technologies and collaborating with diverse teams to push the boundaries of machine learning.

The ideal candidate will have over 4 years of experience, significant knowledge of large-scale model platforms, and proficiency in Python or C++. A competitive salary with comprehensive benefits is offered, including RSUs and a sign-on bonus.