Related skills
data pipeline python pytorch kernel development cuda📋 Description
- Performance profiling and optimization to reduce step time.
- Distributed training pipelines with PyTorch Distributed.
- Kernel development for GPU workloads in Triton or CUDA.
- Data pipeline engineering to maximize training throughput.
🎯 Requirements
- Education: Bachelor's, Master’s, or PhD in CS/CE or related.
- Software Engineering: Strong proficiency in Python.
- ML Frameworks: Extensive hands-on experience with PyTorch.
- ML Knowledge: Experience optimizing training and inference; strong ML concepts.
- Problem Solving: Analytical and data-driven approach to challenges.
🎁 Benefits
- Hybrid option with in-office time in Boston, Pittsburgh, or Las Vegas; fully remote also available.
- Medical, dental, vision, 401k with company match, HSAs, life insurance, and more.
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!