RL Training Framework
xAI· Model
Palo Alto, CA$180k - $440kPosted 3mo ago
About the role
Engineer to develop SpaceXSI's RL training framework, designing systems for RL workloads from ablations to production runs, boosting end-to-end training performance, scalability, and observability.
Responsibilities
- Design and implement the systems backing all RL workloads at SpaceXSI, 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
Qualifications
- 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++
Skills
Required:large-scale distributed systemsPythonJax
Nice to have:RustC++reinforcement learningRL numericsLLM training infrastructure
Benefits
- equity
- comprehensive medical, vision, and dental coverage
- access to a 401(k) retirement plan
- short & long-term disability insurance
- life insurance
- various other discounts and perks
Last updated Oct 21, 2026. Interested? Apply on the company site.
More open roles at xAI
- Supervisor, Child Safety Operations4d agoAustin, TX · Safety
- Security Engineer, Kubernetes4d agoHybrid (Palo Alto, CA; Austin, TX; New York, NY; Washington, DC) · Information Security
- Security Engineer - GRC Sales4d agoInformation Security
- Site Reliability Engineer4d agoHybrid (Memphis/Southaven) · Data Center
- Senior Linux Data Center Technician - Memphis4d agoHybrid (Memphis, TN) · Data Center