Machine Learning Systems Engineer

📋 Description

• Own performance profiling and optimization for ML training with Nsight, PyTorch Profiler.
• Distributed training pipelines with PyTorch Distributed.
• Develop high-performance GPU kernels in Triton or CUDA.
• Optimize data loading pipelines to maximize training throughput.
• Collaborate with researchers to scale large models.

🎯 Requirements

• Bachelor’s, Master’s, or PhD in CS/CE or related field.
• Strong proficiency in Python.
• Extensive PyTorch experience.
• Experience optimizing training and inference; strong ML concepts.
• Excellent analytical and problem-solving skills.

🎁 Benefits

• Medical, dental, and vision coverage.
• 401(k) with company match.
• Health Savings Accounts.
• Life insurance.
• Pet insurance.
• Inclusive and diverse culture.