Machine Learning Systems Research Engineer, Agent Post-training - Enterprise GenAI

Scale AI is a leading AI data foundry focused on accelerating the development of AI applications. The ML Systems Research Engineer will build and optimize algorithms for a next-gen Agent RL training platform, collaborating with teams to enhance model training and deployment for various enterprise applications.

• Build, profile, and optimize training and inference framework
• Post-train state-of-the-art models to define stable post-training recipes for enterprise engagements
• Collaborate with ML teams to accelerate research and development
• Create a next-gen agent training algorithm for multi-agent/multi-tool rollouts

• 1-3 years of LLM training in a production environment
• Passionate about system optimization
• Experience with post-training methods like RLHF/RLVR and algorithms like PPO/GRPO
• Knowledge of modern GPU cluster architecture
• Experience with multi-node LLM training and inference
• Strong software engineering skills: CUDA, PyTorch, transformers, flash attention
• Strong written and verbal communication skills
• PhD or Masters in Computer Science or related field