Machine Learning Performance Engineer
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
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.
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
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
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