Latest AI JobsBrowse Jobs
Scale AI logo

ML Research Engineer, ML Systems

Scale AISan Francisco, CA; Seattle, WA; New York, NYResearch
Apply on company site
Salary
Not listed
Job type
Full-time
Posted
10 Mar 2025
Status
Open · checked 7d ago

About this role

The ML Research Engineer, ML Systems position at Scale AI is an exciting opportunity based in San Francisco, Seattle, or New York to build foundational frameworks for large language model training and inference. This role suits professionals passionate about distributed machine learning systems and optimizing performance for next-generation artificial intelligence applications.

Role Overview

Scale AI is hiring an ML Research Engineer to join the ML platform team, known as RLXF. This team develops the internal distributed framework that powers machine learning engineers, researchers, and data scientists for rapid model training, evaluation, and data curation. The professional in this position will collaborate across multiple internal groups to shape the future of artificial intelligence through fundamental framework innovations and advanced platform optimization.

Key Responsibilities

Build, profile, and optimize the internal training and inference framework.
Partner closely with machine learning teams to accelerate research, development, and data curation workflows for upcoming model generations.
Research and integrate cutting-edge technologies to continuously improve machine learning system performance.

Requirements

Strong passion for system optimization and distributed architecture.
Practical background in multi-node large language model training and inference.
Demonstrated ability to develop large-scale distributed machine learning systems.
Robust software engineering background with proficiency in tools and frameworks such as PyTorch, CUDA, transformers, and flash attention.
Exceptional written and verbal communication skills with the capability to thrive in a cross-functional team environment.

Nice to Have

Proven expertise in post-training techniques and next-generation large language model use cases, including instruction tuning, reinforcement learning from human feedback, tool use, reasoning, agents, and multimodal systems.

Skills and Tools

PyTorch
Large Language Models (LLMs)
CUDA
Transformers
Flash Attention
Distributed Systems

Salary and Benefits

Eligible full-time employees in San Francisco, New York, and Seattle receive a base salary range between $189,600 and $237,000 USD. Compensation packages also include equity grants subject to board approval. Comprehensive benefits feature health, dental, and vision coverage, retirement plans, a learning and development stipend, generous paid time off, and potential commuter stipends.

About the Company

Scale AI focuses on building reliable artificial intelligence systems to drive major decision-making processes. The organization delivers high-quality data and comprehensive full-stack technologies utilized by prominent enterprises, medical institutions, and government agencies. Renowned partners and clients include Meta, Mayo Clinic, Ernst & Young, and multiple United States defense branches. Scale AI operates as an inclusive, equal opportunity employer dedicated to accessible workplaces.

How to Apply

Interested candidates should visit the official Scale AI careers page to submit their application for the ML Research Engineer role.

Skills and tech stack

Questions candidates ask

What locations are available for this role?

This position is located in San Francisco, CA; Seattle, WA; and New York, NY.

What is the salary range for the ML Research Engineer position?

The base salary range for this full-time position in San Francisco, New York, and Seattle is $189,600 to $237,000 USD.

What technical skills are required?

Candidates need strong software engineering skills with proficiency in PyTorch, CUDA, transformers, flash attention, and experience with large-scale distributed ML systems and multi-node LLM training.

What benefits are provided by Scale AI?

Eligible employees receive comprehensive health, dental, and vision coverage, retirement benefits, learning and development stipends, generous paid time off, commuter stipends, and equity-based compensation.

How can I apply for this job?

Candidates should visit the official Scale AI careers page to complete and submit their application.

Ready to apply?

The application is on the official Scale AI careers page.

Apply on company site