Member of Technical Staff ML Performance

You will optimize machine-learning systems for throughput and latency. You will contribute to open-source projects and container runtime performance work, improve language and diffusion model execution, and diagnose GPU performance bottlenecks across algorithms and host overhead.

Responsibilities

• Improve machine-learning system performance at scale
• Contribute to open-source projects and container runtime development
• Optimize language and diffusion model throughput and latency
• Debug GPU performance bottlenecks
• Improve algorithm efficiency and reduce host overhead

Requirements

• 5+ years of experience writing high-quality high-performance code
• Experience with Torch, high-level ML frameworks, and inference engines
• Familiarity with Nvidia GPU architecture and CUDA
• Experience with machine-learning performance engineering

Benefits

• Equity