Machine Learning Engineer (Reinforcement Learning)

1 Month ago • All levels
Research Development

Job Description

We are seeking a Machine Learning Engineer to develop and scale distributed reinforcement learning systems for model training. This role involves deploying elastic environment microservices, designing effective reward systems, and optimizing multi-node and multi-datacenter training pipelines to enhance performance and efficiency.
Good To Have:
  • History of OSS contributions
  • Knowledge of TorchTitan and SGLang or vLLM
Must Have:
  • Designing and implementing RL pipelines from reward modeling to policy optimization
  • Optimizing RL training stability and sample efficiency for large models
  • Verifying numerical correctness across inference and training
  • Performance engineering on trainer-inference communication
  • Validating methods from recent publications
  • Hands-on experience with reinforcement learning in production systems
  • Deep understanding of policy-space methods (GRPO, PPO, etc.)
  • Experience profiling distributed systems

Add these skills to join the top 1% applicants for this job

game-texts
reinforcement-learning
microservices
machine-learning

We’re looking for an MLE to build and scale distributed reinforcement learning systems for model training. You’ll deploy elastic environment microservices, design reward systems and optimize multi-node and multi-datacenter training pipelines.

Responsibilities:

  • Designing and implementing RL pipelines from reward modeling to policy optimization
  • Optimizing RL training stability and sample efficiency for large models
  • Verifying numerical correctness across inference and training
  • Performance engineering on trainer-inference communication
  • Validating methods from recent publications

Qualifications:

  • Hands-on experience with reinforcement learning in production systems
  • Deep understanding of policy-space methods (GRPO, PPO, etc.)
  • Experience profiling distributed systems

Preferred:

  • History of OSS contributions
  • Knowledge of TorchTitan and SGLang or vLLM

Set alerts for more jobs like Machine Learning Engineer (Reinforcement Learning)
Set alerts for new jobs by Nousresearch
Set alerts for Research Development (Remote) jobs

Contact Us
hello@outscal.com
Made in INDIA 💛💙