xAI is looking for an engineer to join its RL infrastructure team to design, implement, and optimize the reinforcement learning training framework, supporting workloads from small-scale ablations to production training runs.
Member of Technical Staff - RL Training Framework at xAI
On-site - Palo Alto, CA, United States
More jobs at xAISalary
USD 180,000 - 440,000
Requirements
Skills
- Experience building, debugging, and optimizing large-scale distributed systems
- Comfortable diving into unfamiliar areas and solving problems at all levels of the stack
- Proficiency in Python, Jax, Rust, or C++
- Experience with large scale LLM training infrastructure
- Strong knowledge of reinforcement learning techniques
- Experience with RL numerics
Responsibilities
- 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
Technologies
PythonJaxRustC++
See if your resume is ready for this job
See how our AI can optimize your resume and improve your chances for this role.