Skip to main content

Scale AI

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

San Francisco, CA; New York, NYFrom $331kmidAdded 1 month ago

About this role

As a Machine Learning Systems Research Engineer at Scale, you'll develop advanced algorithms for agent training and optimize ML systems for enterprise applications. Your work will support cutting-edge technologies across a variety of sectors, including cybersecurity and health tech.

What you'll do

  • Build and optimize training and inference frameworks
  • Post-train state-of-the-art models for enterprise use
  • Collaborate with ML teams to assist their research and development
  • Develop next-gen agent training algorithms for multi-agent systems

What they're looking for

  • 1-3 years of experience in LLM training
  • Knowledge of system optimization
  • Familiarity with post-training methods like RLHF/RLVR
  • Experience with multi-node LLM training
  • Strong software engineering skills (CUDA, Pytorch)
  • Excellent communication skills
  • PhD or Masters in Computer Science or related field

Benefits

  • Comprehensive health, dental, and vision coverage
  • Retirement benefits
  • Learning and development stipend
  • Generous PTO
  • Commuter stipend
Apply with Autofill

Opens the application — the Jobs AI extension fills it for you. Set up autofill

Opens the official application on the employer’s site. No login required.

Scale AI

Scale AI builds a Generative AI Data Engine and ML infrastructure platforms that power LLM training, evaluation, and production serving at scale, along with data solutions for robotics and autonomous driving. The company is hiring Software Engineers for full-stack feature development and infrastructure systems, identity/security specialists for platform engineering, and Solutions Engineers to support enterprise clients and pre-sales processes.

Website
scale.com
View all jobs at Scale AI