Related skills
python pytorch data pipelines cuda triton📋 Description
- Profiling and optimizing data loading, gradients, and communication.
- Distributed training pipelines with PyTorch Distributed.
- Design and maintain high-performance GPU kernels in Triton or CUDA.
- Data pipeline engineering to maximize training throughput.
🎯 Requirements
- Education: Bachelor's, Master's, or PhD in CS/CE or related field.
- Python: Strong proficiency in Python.
- ML Frameworks: Extensive PyTorch experience.
- ML Knowledge: Optimizing training/inference; solid ML concepts.
- Problem Solving: Analytical, data-driven approach.
🎁 Benefits
- Hybrid schedule with in-office time in Boston, Pittsburgh, or Las Vegas, or fully remote.
- Motional benefits: medical, dental, vision, 401k with match.
Meet JobCopilot: Your Personal AI Job Hunter
Automatically Apply to Engineering Jobs. Just set your
preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.
Help us maintain the quality of jobs posted on Empllo!
Is this position not a remote job?
Let us know!