ML Research Engineer, ML Systems
Scale AISan Francisco, California, United States • Seattle, Washington, United States • New York, New York, United States
Posted March 10, 2025
200800 - 251000 USD year
Full Time
About the Company
Scale accelerates AI development with tools for high-quality data curation and generative model customization.
Job Description
Scale’s ML platform team builds a distributed framework for large language model training and inference. The platform powers MLEs, researchers, data scientists, and operators for fast and automatic training and evaluation of LLMs. The ML Research Engineer will build and optimize the platform to enable next-generation LLM training, inference, and data curation.
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 and tools such as CUDA, Pytorch, transformers, flash attention, etc.
- Strong written and verbal communication skills and the ability to operate in a cross-functional team environment
Benefits
- Comprehensive health, dental, and vision coverage
- Retirement benefits
- Learning and development stipend
- Generous PTO
- Commuter stipend (optional)
