ML Research Engineer, ML Systems
Scale AI
12 months ago
New York, NY, USA +2 more
Mid Level / Senior
H1B Sponsor
Base Salary
$201k - $251k/yr
Responsibilities
- Build, profile, and optimize the training and inference framework.
- Collaborate with ML teams to accelerate research and development.
- Enable the development of next-generation models and data curation.
- Research and integrate state-of-the-art technologies to optimize the ML system.
Requirements
- Strong excitement about system optimization.
- Experience with multi-node LLM training and inference.
- Experience with developing large-scale distributed ML systems.
- Strong software engineering skills, proficient in frameworks like CUDA, Pytorch, and transformers.
- Strong written and verbal communication skills.
Benefits
- Comprehensive health, dental, and vision coverage.
- Retirement benefits.
- Learning and development stipend.
- Generous PTO.
- Potential commuter stipend.
Tech Stack
PyTorch
Categories
AI & MLData Engineering