# ML Research Engineer, ML Systems

**Company:** [Scale AI](https://hotfix.jobs/companies/scale-ai)
**Location:** San Francisco, CA, Seattle, WA, New York, NY
**Role:** ML Engineering
**Salary:** $218k – $273k/yr
**Skills:** PyTorch, CUDA, Transformers, Flash Attention, Llm Training, Distributed Ml Systems, RLHF, Multi-Node Training, Ml Inference, System Optimization
**Posted:** 2026-02-12

> Builds and optimizes distributed frameworks for LLM training and inference on Scale's RLXF platform. Collaborates with ML teams to accelerate research, requiring expertise in PyTorch, CUDA, transformers, and large-scale distributed systems.

## Job Description

## You will:
- Build, profile and optimize our training and inference framework
- Collaborate with ML teams to accelerate their research and development and enable them to develop the next generation of models and data curation
- Research and integrate state-of-the-art technologies to optimize our ML system

## Ideally you’d have:
- 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

## Nice to haves:
- Demonstrated expertise in post-training methods or next generation use cases for large language models including instruction tuning, **RLHF**, tool use, reasoning, agents, and multimodal, etc.

## Similar roles

- [ML Systems Engineer, Robotics](https://hotfix.jobs/jobs/fd6494c5-b63a-412e-87fe-48b825f54495) - Scale AI - San Francisco, CA - $218k – $273k/yr
- [Machine Learning Research Engineer, Agent Data Foundation - Enterprise GenAI](https://hotfix.jobs/jobs/c04650e4-6722-4170-ae84-943c4d2aac4f) - Scale AI - San Francisco, CA - $218k – $273k/yr
- [AI Engineer, Applied ML](https://hotfix.jobs/jobs/fa2a14c5-a9f8-41ed-b414-a6dc50f0126d) - Perplexity - San Francisco, CA - $220k – $405k/yr
- [AI Inference Engineer](https://hotfix.jobs/jobs/76655dad-b39d-48cd-8de2-290b953d6d0c) - Perplexity - San Francisco, CA - $220k – $405k/yr
- [Software Engineer, Enterprise AI](https://hotfix.jobs/jobs/98385fe7-6f4c-4386-ab6e-d4033a6d2440) - Scale AI - New York, NY - $216k – $270k/yr

**Apply:** https://hotfix.jobs/jobs/c814c5a3-7e79-4c03-8f4b-89eb8352da30
**Canonical:** https://hotfix.jobs/jobs/c814c5a3-7e79-4c03-8f4b-89eb8352da30