Machine Learning Backend Engineer Graduate (AML MLDev) - 2027 Start
Summary
Develops high-performance ML kernels and optimizes model compilation for faster inference/training in a large-scale AI product. Works across full-stack performance, collaborating with algorithm and business teams to reduce bottlenecks and improve computational efficiency.
- Analyze full stack performance
- Build model compilation system
- Collaborate with algorithm and business teams
- Design and develop high performance kernel libraries
- Develop optimization solutions
- Diagnose bottlenecks
- Fuse kernels
- Generate code
- Implement general purpose and custom kernels
- Improve computing utilization
- Optimize model compilation
- Optimize model computational performance
- Perform graph optimization
- Schedule computation
- Tune inference and training bottlenecks
Perks/Benefits:
- 401k match
- Dental insurance
- Life insurance
- Medical insurance
- Paid Holidays
- Paid parental leave
- Paid personal time
- Paid sick days
- Vision insurance
- Wellbeing benefits