Quick Overview
Job Description
About the Role
Join an early-stage AI team building reinforcement learning systems to help AI agents handle complex work. You will own the full loop from environment design through training and evaluation, while helping shape the team's technical direction.
What You'll Do
Design and build reinforcement learning environments, reward functions, and training pipelines.
Train and fine-tune models using methods such as PPO, GRPO, DPO, RLHF, and RLAIF.
Build evaluation frameworks to measure model and agent performance.
Run experiments, interpret results, and use findings to guide what to try next.
Scale training on GPU clusters and maintain reliable training pipelines.
Turn research ideas into production systems and help establish engineering practices.
Contribute to hiring as the engineering team grows.
What We're Looking For
At least 2 years of hands-on experience in reinforcement learning or machine learning engineering.
Strong Python skills and deep experience with PyTorch or JAX.
Practical experience training models with reinforcement learning methods, including policy gradients, reward modeling, or RLHF.
Experience with distributed training, GPU infrastructure, RL environments, reward functions, training pipelines, and evaluation frameworks.
A track record of bringing research ideas into production, plus a degree in computer science, mathematics, physics, or a related field.
Experience with LLM post-training, agent training, simulation frameworks, or training libraries is a plus.
Compensation & Benefits
Compensation is $125,000 to $200,000 USD annually. Visa sponsorship is available.
Location
This is an on-site role based in San Francisco, California, United States.
Similar jobs
- SA
Machine Learning Research Intern (Summer 2027)
NewAuto ApplyScale AI
New York🇺🇸Hybrid12 hours agoSpringMachine LearningNLP+4Technology - NA
Director of Bioinformatics and Machine Learning
NewAuto ApplyNatera
San Carlos🇺🇸Hybrid12 hours agoMachine LearningIntellectual PropertyTechnology - CL
Founding Engineer (AI/ML)
NewAuto ApplyClera
San Francisco🇺🇸On-site14 hours agoSupabaseMachine LearningLLM+7Technology - CL
AI/ML Engineer, Individual Contributor
NewAuto ApplyClera
San Francisco🇺🇸$120k - $200k/yrOn-site14 hours agoLLMTechnology - TH
Research, Tinker, RL Systems
NewAuto ApplyThinkingmachines
San Francisco🇺🇸On-site7 hours agoMachine LearningDeep LearningLLM+3 - CH
Senior Machine Learning Engineer
NewAuto ApplyCheckr
Denver🇺🇸Hybrid14 hours agoRubyMLOpsMLflow+9Technology - SO
Member of Technical Staff (ML)
NewAuto ApplySolstice
New York City🇺🇸$200k - $300k/yrOn-site4 hours agoMachine LearningConcreteLLM+1 - PM
Member of Technical Staff - Research Engineer, Post-training
NewAuto ApplyPreference Model
San Francisco🇺🇸On-site9 hours agoMachine LearningLLMPyTorch+1Engineering - PM
Member of Technical Staff - AI/ML Infrastructure Engineer, RL Post-training
NewAuto ApplyPreference Model
San Francisco🇺🇸On-site9 hours agoGCPAWSKubernetes+3Technology - PM
Member of Technical Staff - Research Engineer, RL Environments
NewAuto ApplyPreference Model
San Francisco🇺🇸On-site9 hours agoMachine LearningCUDALLM+2Engineering - NO
AI/ML Engineer (Midlevel)
NewAuto ApplyNode.Digital
Herndon, Virginia🇺🇸Hybrid12 hours agoSOAPSQLSQL Server+19Technology - NO
Junior AI/ML Engineer
NewAuto ApplyNode.Digital
Herndon, Virginia🇺🇸Hybrid12 hours agoMLflowMachine LearningNLP+12Technology