About the Role Join an early-stage AI team building reinforcement learning systems from the ground up. You will own the end-to-end process, from environment design and model training to evaluation, helping shape the technical direction and engineering culture. 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. Develop evaluation frameworks to measure model and agent performance. Run experiments, interpret results, and decide what to explore next. Scale training on GPU clusters and keep training pipelines reliable. Turn research ideas into production systems. Help establish engineering practices and hire future team members. 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 model or agent evaluation. A track record of taking research ideas into production systems. A degree in computer science, mathematics, physics, or a related field; a master's or PhD is a plus. LLM post-training or agent training experience is beneficial, as are RL publications or open-source work and experience with simulation frameworks or training libraries. Compensation & Benefits Base salary: $125,000 to $200,000 USD annually. Visa sponsorship is available. Location On-site in San Francisco, California, United States. Candidates must be based in or willing to relocate to San Francisco.
Similar jobs
Job
AI Scientist, Product Lifecycle Management (PLM)
Avathon
AI/MLHuggingfaceLlm
+1
$160,000–$240,000/yr
gross
🏢 On-site
Pleasanton
🗣️ EN
Job
AI Scientist, Platform
Avathon
AI/MLDatabricksHuggingfaceLlm
+2
$160,000–$240,000/yr
gross
🏢 On-site
Pleasanton
🗣️ EN

Job
ML Algorithm Mapping and Performance Engineer, Core ML
Cerebras
AI/MLCppLlm
💰 Salary: not specified
🔀 Hybrid
Sunnyvale
🗣️ EN

Job
ML Runtime and Kernel Engineer - Core ML
Cerebras
AI/MLCppLlm
💰 Salary: not specified
🔀 Hybrid
Sunnyvale
🗣️ EN

Job
AI TLM Performance Modeling
Qualcomm
AI/MLCppLlm
+1
$162,000–$243,000/yr
gross
🏢 On-site
San Diego
🗣️ EN

Job
Eloquent AI Fellowship Program
Eloquentai
AI/MLData ScienceLlm
+1
💰 Salary: not specified
🏢 On-site
San Francisco
🗣️ EN
