Research Engineer, Infrastructure, Kernels

What You’ll Do

Skills and Qualifications

• Minimum qualifications:
• Bachelor’s degree or equivalent experience in computer science, electrical engineering, statistics, machine learning, physics, robotics, or similar.
• Strong engineering skills, ability to contribute performant, maintainable code and debug in complex codebases
• Understanding of deep learning frameworks (e.g., PyTorch, JAX) and their underlying system architectures.
• Thrive in a highly collaborative environment involving many, different cross-functional partners and subject matter experts.
• A bias for action with a mindset to take initiative to work across different stacks and different teams where you spot the opportunity to make sure something ships.
• Proficiency in CUDA, CuTe, Triton, or other GPU programming frameworks.
• Demonstrated ability to analyze, profile, and optimize compute-intensive workloads.
• Preferred qualifications — we encourage you to apply if you meet some but not all of these:
• Experience training or supporting large-scale language models with tens of billions of parameters or more.
• Track record of improving research productivity through infrastructure design or process improvements.
• Experience developing or tuning kernels for deep learning frameworks such as PyTorch, JAX, or custom accelerators.
• Familiarity with tensor parallelism, pipeline parallelism, or distributed data processing frameworks.
• Experience implementing low-precision formats (FP8, INT8, block floating point) or contributing to related compiler stacks (e.g., XLA, TVM).
• Contributions to open-source GPU, ML systems, or compiler optimization projects.
• Prior research or engineering experience in numerical optimization, communication-efficient training, or scalable AI infrastructure.

Logistics