ML Systems Engineer, ML Acceleration

📋 Description

• Profile and optimize data loading, gradient computation, and communication.
• Optimize distributed training pipelines with PyTorch Distributed.
• Design and maintain high-performance GPU kernels in Triton or CUDA.
• Build robust data loading pipelines to maximize training throughput.
• Work at the intersection of ML research and high-performance systems.

🎯 Requirements

• Education: Bachelor’s, Master’s, or PhD in CS, CE, or related technical discipline.
• Software Engineering: Strong proficiency in Python.
• ML Frameworks: Extensive hands-on experience with PyTorch.
• ML Knowledge: Experience optimizing model execution during training and inference; strong ML concepts.
• Problem Solving: Excellent analytical and data-driven approach.

🎁 Benefits

• Medical, dental, and vision coverage.
• 401(k) with company match.
• Health Savings Account program.
• Life insurance and disability coverage.
• Pet insurance and more benefits.
• Inclusive, diverse culture.