Drimva

Scale AI

ML Research Engineer, ML Systems

$190K–$237K

San Francisco, CA +3 · Full-time · Posted Mar 10, 2025

Location
San Francisco, CA
Seattle, WA
New York, NY
San Francisco, California, United States

About the job

Build and optimize Scale’s distributed ML platform for large language model training, inference, evaluation, and data curation.

What you'll do

  • Build, profile, and optimize training and inference frameworks.
  • Collaborate with ML teams to accelerate research and develop models and data curation capabilities.
  • Research and integrate technologies to optimize ML systems.

What they ask for

  • Experience with multi-node LLM training and inference.
  • Experience developing large-scale distributed ML systems.
  • Strong software engineering skills and proficiency with relevant frameworks and tools.
  • Strong written and verbal communication skills; ability to work cross-functionally.

Nice to have

  • Expertise in LLM post-training methods or next-generation use cases such as instruction tuning, RLHF, tool use, reasoning, agents, and multimodal applications.

Pay

$189,600 — $237,000 USD (base salary)

Tools

CUDA, Pytorch, transformers, flash attention

AI focus

Develop and optimize systems for LLM training, inference, evaluation, and data curation; integrate state-of-the-art ML technologies.

Model-written summary · the original posting is authoritative

First seen Oct 10, 2026 · Last verified Nov 4, 2026 (UTC)