Scale AI · 2 hours ago
Machine Learning Research Engineer, Enterprise ML Systems
Scale AI is a leading AI data foundry focused on accelerating the development of AI applications. The Machine Learning Research Engineer will build algorithms for a next-gen Agent RL training platform, support large scale training, and integrate state-of-the-art technologies to optimize ML systems for enterprise clients.
Artificial Intelligence (AI)Data Collection and LabelingGenerative AIImage RecognitionMachine Learning
Responsibilities
Build, profile and optimize our training and inference framework
Post-train state of the art models, developed both internally and from the community, to define stable post-training recipes for our enterprise engagements
Collaborate with ML teams to accelerate their research and development, and enable them to develop the next generation of models and data curation
Create a next-gen agent training algorithm for multi-agent/multi-tool rollouts
Qualification
Required
At least 1-3 years of LLM training in a production environment
Passionate about system optimization
Experience with post-training methods like RLHF/RLVR and related algorithms like PPO/GRPO etc
Ability to demonstrate know-how on how to operate the architecture of the modern GPU cluster
Experience with multi-node LLM training and inference
Strong software engineering skills, proficient in frameworks and tools such as CUDA, Pytorch, transformers, flash attention, etc
Strong written and verbal communication skills to operate in a cross functional team environment
PhD or Masters in Computer Science or a related field
Benefits
Comprehensive health, dental and vision coverage
Retirement benefits
A learning and development stipend
Generous PTO
A commuter stipend
Company
Scale AI
Scale’s mission is to develop reliable AI systems for the world’s most important decisions.
H1B Sponsorship
Scale AI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (82)
2024 (54)
2023 (29)
2022 (17)
2021 (10)
2020 (10)
Funding
Current Stage
Late StageTotal Funding
$15.9BKey Investors
MetaAccelTiger Global Management
2025-06-10Corporate Round· $14.3B
2025-06-04Series Unknown
2024-05-21Series F· $1B
Recent News
2025-12-31
Investing.com
2025-12-30
Company data provided by crunchbase