Research Engineer - LLM Post-Training and RL Environment Development

preference model • United State
Visa Sponsorship Relocation
Apply
AI Summary

Develop and optimize RL training environments for large language models, focusing on self-directed learning and post-training. Implement novel RL methodologies and scale infrastructure for distributed experiment management. Requires strong Python, PyTorch/JAX, and RL framework expertise with a blend of research and engineering rigor.

Key Highlights
Build RL environments reflecting real-world complexity for LLM post-training
Implement and optimize RL training infrastructure using Verl/OpenRLHF
Profile and optimize end-to-end training pipelines for maximum throughput
Experience with LLM post-training pipelines and reward signal design
No PhD required; adaptability and communication skills prioritized
Key Responsibilities
Train and evaluate models on proprietary RL environments to validate data quality and surface gaps in task coverage
Architect and optimize RL training infrastructure from training abstractions to distributed experiment management
Design, implement, and test training environments, evaluations, and methodologies for RL agents
Profile and optimize training runs end-to-end from data loading through reward computation to maximize experiment throughput
Help scale systems to handle increasingly complex research workflows
Technical Skills Required
Python PyTorch JAX RL training frameworks ML infrastructure Data loading Reward computation Distributed experiment management
Benefits & Perks
Competitive cash and equity compensation (>90th percentile)
Ownership and autonomy
Health, vision, dental benefits
401K match
Lunch provided everyday onsite
Weekly snack orders
Visa sponsorship
Relocation support available
Nice to Have
Experience evaluating model outputs and building reward or evaluation signals
Staying current on post-training research and translating papers into running code
Strong opinions about structuring RL training code for reproducibility and fast iteration
Balancing research exploration with engineering rigor
Strong systems design and communication skills

Job Description


About Us

Preference Model is building automated ML research engineering.

Existing frontier models are brittle when applied to real-world ML tasks. The present bottleneck is the lack of high-quality RL training environments. Our first step is to build RL environments that reflect real-world complexity, with diverse tasks and robust reward functions.

Our founding team has previous experience on Anthropic’s data team building data infrastructure, and datasets behind Claude. We are partnering with leading AI labs to push AI closer to achieving its transformative potential.

About The Role

Models of the future will be able to train themselves on tasks that they are not good at. We are interested in investigating how far we can push the boundaries of self-directed learning. We are looking for Research Engineers or Research Scientists to push the frontier of post-training on large language models in a role that blends research and engineering, requiring you to implement novel approaches and shape research directions.

What You Will Do:

  • Train and evaluate models on our proprietary RL environments to validate data quality, surface gaps in task coverage, and close the feedback loop between environment design and model capability.
  • Architect and optimize our RL training infrastructure, from training abstractions to distributed experiment management, using frameworks like Verl, OpenRLHF, or similar. Help scale our systems to handle increasingly complex research workflows.
  • Design, implement, and test training environments, evaluations, and methodologies for RL agents.
  • Profile and optimize training runs end-to-end, from data loading through reward computation, to maximize experiment throughput and shorten the research iteration cycle.

What We are Looking For

  • Experience running end-to-end LLM post-training pipelines
  • Proficiency in Python and PyTorch or JAX
  • Experience with at least one modern RL training framework
  • Experience building and operating ML infrastructure at scale

You may be a good fit if you also:

  • Have experience evaluating model outputs and building reward or evaluation signals
  • Stay current on post-training research and can translate papers into running code
  • Have strong opinions (loosely held) about how to structure RL training code for reproducibility and fast iteration
  • Can balance research exploration with engineering rigor
  • Have strong systems design and communication skills

Candidates don't need a PhD or extensive publications. Some of the best researchers have no formal ML training and gained experience building industry products. We believe adaptability combined with exceptional communication and collaboration skills are the most important ingredients for successful startup research.

What we offer:

  • Competitive cash and equity compensation (>90th percentile)
  • Ownership and autonomy in a fast moving startup environment
  • Opportunity to work with top machine learning engineers
  • Health, vision, dental, benefits
  • 401K match
  • Lunch provided everyday onsite
  • Weekly snack orders
  • Visa sponsorship & relocation support available

We value diverse perspectives and experiences. If you're excited about this role but don't check every box, we still encourage you to apply.

Note: We utilize AI note-taking during our interview sessions to ensure we capture all answers and details accurately. Candidates are allowed to use AI note-takers as well, however, no other AI tools are permitted during any live interviews.

Compensation Range: $200K - $350K


Similar Jobs

Explore other opportunities that match your interests

HR Operations Specialist

Programming
•
3m ago
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Associate

food lion

United State

Reverse Engineer Researcher

Programming
•
23m ago

Premium Job

Sign up is free! Login or Sign up to view full details.

•••••• •••••• ••••••
Job Type ••••••
Experience Level ••••••

Jobs via Dice

United State

Embedded Software Architect 5 - Missile Systems Development

Programming
•
42m ago

Premium Job

Sign up is free! Login or Sign up to view full details.

•••••• •••••• ••••••
Job Type ••••••
Experience Level ••••••

Northrop Grumman

United State

Subscribe our newsletter

New Things Will Always Update Regularly