Member of Technical Staff; RL Training Framework

• The RL infrastructure team is looking for an engineer to help develop our RL training framework
• Design and implement the systems backing all RL workloads at xAI, from small scale ablations to production training runs
• Profile, debug, and optimize end-to-end training performance
• Improve scalability and observability of the RL stack

• Health and wellness:
  Comprehensive health insurance including medical, dental, vision, and disability coverage
• Life and family:
  Life and AD&D insurance and fertility benefits to ensure our team’s well-being and peace of mind
• Flexible vacation:
  We work hard but avoid burn out. Take time off when you need it
• Visa sponsorship:
  We support international talent with visa sponsorship to join our team
• 401(k) plan:
  Retirement savings plan to secure your financial future

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.