About this role

Join xAI's RL infrastructure team to develop and optimize the core training framework for AI systems. You will design, implement, and scale systems for all RL workloads, ensuring high performance and observability. This role requires strong programming skills and a passion for solving complex problems in distributed systems.

What we are looking for

6

Design and implement RL training framework systems
Optimize end-to-end training performance
Improve scalability and observability of RL stack
Experience with large-scale distributed systems
Proficiency in Python, Jax, Rust, or C++
Experience with LLM training infrastructure

Skills mentioned

3

PythonC++Rust