Member of Technical Staff - RL Inference

Experience in building, debugging, and optimizing efficiency of large-scale distributed systems

Proficiency in programming languages such as Python, C++ and/or Rust; frameworks such as PyTorch, Jax, CUDA

Experience in developing inference engines, e.g.

For details on data processing, view our

Design and optimize our inference stack for all shapes of RL workloads at xAI, from small scale ablations to production training runs.

Analyze, profile and address performance bottlenecks in large scale RL systems

Work closely with the modelling team to efficiently implement novel RL techniques and algorithms