GenAI Systems Research Engineer - Post-Training

Scale AI, Inc.; New York (NY); On-site; USD 265,000 - 331,000; Full time

Job summary

Scale AI, Inc. is hiring a Machine Learning Systems Research Engineer/Scientist for Enterprise GenAI post-training work in New York, NY.

You will build and optimize training and inference pipelines, post-train state-of-the-art models, and collaborate with ML teams to push the frontier of agent-based training used by enterprise clients. The role requires strong software engineering skills, experience with CUDA, PyTorch, Transformers, and Flash attention, and a track record of multi-node LLM

Qualifications

• At least 1-3 years of LLM training in a production environment.

Responsibilities

• Build, profile and optimize our training and inference framework.

Skills

LLM training

Post-training methods

RLHF/RLVR

Multi-node training

GPU cluster

System optimization

Distributed training

Software engineering

Communication

Education

PhD or Masters in CS or related

Tools

CUDA

PyTorch

Transformers

Flash attention

Job description

Scale AI, Inc. is hiring a Machine Learning Systems Research Engineer/Scientist for Enterprise GenAI post-training work in New York, NY.

You will build and optimize training and inference pipelines, post-train state-of-the-art models, and collaborate with ML teams to push the frontier of agent-based training used by enterprise clients. The role requires strong software engineering skills, experience with CUDA, PyTorch, Transformers, and Flash attention, and a track record of multi-node LLM