Machine Learning Performance Engineer

Work in a highly technical ML Performance and AI Acceleration team to improve training and inference performance through compiler development, low-level systems optimization, and hardware architecture design.

Responsibilities

• Increase the performance of training and inference platforms
• Build and maintain an optimizing compiler for accelerated computing platforms
• Work with memory hierarchies and degrees of parallelism
• Analyze microarchitectural nuances in instruction dispatch rates and irregular bitwidth numerics
• Assist in designing novel hardware architectures for accelerating AI models
• Map AI models from JAX graphs to transistors

Requirements

• Experienced software engineer, likely in C, C++, Rust, or a similar language
• Confident solving complex algorithmic problems
• Understanding of and experience targeting GPUs, TPUs, or similar hardware
• Experience with data-flow-oriented compilers for AI or scientific computing
• Experience writing Linux kernel modules or other operating-system development

Benefits

• Onsite gym, sauna, and fitness classes at no charge
• Extensive medical benefits including an onsite doctor and therapist at no charge
• Breakfast and lunch provided daily
• Support for caregivers, including emergency dependent care
• 25 days paid holiday plus statutory holidays and paid sick days
• Generous pension contributions