ML Systems Engineer, ML Acceleration

📋 Description

• Support researchers to train frontier models at scale.
• Focus on speed, cost, reliability, and throughput.
• Work at the intersection of ML research and high-performance systems.
• Profile bottlenecks in data loading, gradient compute, and communication.
• Develop GPU kernels in Triton or CUDA and optimize data paths.
• Build robust data loading pipelines to maximize throughput.

🎯 Requirements

• Bachelor’s, Master’s, or PhD in CS, Computer Engineering, or related field.
• Strong proficiency in Python.
• Extensive hands-on PyTorch experience.
• Experience optimizing ML model execution during training and inference.
• Strong analytical and problem-solving skills; data-driven approach.

🎁 Benefits

• Medical, dental, and vision benefits.
• 401(k) with company match and health savings accounts.
• Life insurance and pet insurance.
• Diversity, inclusion, and a people-first culture.
• Hybrid or remote-friendly work environment.