Member of Technical Staff - RL Training Framework

Highlights

• Build systems backing all RL workloads from ablations to production.
• Optimize end-to-end training performance and scalability.
• Improve observability across the RL training stack.

What they offer

• equity
• comprehensive medical, vision, and dental coverage
• 401(k) retirement plan
• short & long-term disability insurance
• life insurance

Ideal candidate

A curious, hands-on engineer who thrives on hard problems and owns systems end to end. You care about engineering excellence, fast iteration, and scaling RL training reliably.

Requirements

• Experience building, debugging, and optimizing efficiency of large-scale distributed systems
• Comfortable diving into unfamiliar areas and solving problems at all levels of the stack
• Proficiency in Python, Jax, Rust, and/or C++
• Experience with large scale LLM training infrastructure
• Strong knowledge of reinforcement learning techniques
• Experience with RL numerics