2 days ago
Base Salary
$125k - $200k/yr
Responsibilities
- Design and build reinforcement learning environments, reward functions, and training pipelines.
- Train and fine-tune models using PPO, GRPO, DPO, RLHF, and RLAIF.
- Develop evaluation frameworks for model and agent performance.
- Run experiments, interpret results, and determine future research directions.
- Scale training on GPU clusters and maintain reliable training pipelines.
- Translate research ideas into production systems.
- Help establish engineering practices and hire future team members.
Requirements
- 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 with 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.
- Preferred qualifications include LLM post-training or agent training experience, RL publications or open-source work, and experience with simulation frameworks or training libraries.
Benefits
- Base salary of $125,000 to $200,000 USD annually.
- Visa sponsorship is available.
- On-site in San Francisco, California; candidates must be based in or willing to relocate to San Francisco.
Categories
AI ResearchML Engineering
