Member of Technical Staff (RL Training Framework)
xAI
RemoteFull timeMid level$180k – $440kPosted today
Apply with JobAssistAbout the role
- 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 SpaceXAI, 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
Benefits
- 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
- Proficiency in Python, Jax, Rust, and/or C++
- Comfortable diving into unfamiliar areas and solving problems at all levels of the stack
- Experience with large scale LLM training infrastructure
- Experience with RL numerics
- Strong knowledge of reinforcement learning techniques
Millions of jobs, with real people getting hired every day
20,000+
New jobs added daily7,000,000+
Verified job listings500,000+
Tailored applications submittedFAQ
Questions, answered
Click "Apply with JobAssist" – we tailor your resume and application to this role and submit it for your approval.
Yes. This role at xAI was screened before publishing – we confirmed the employer before listing it.
The employer didn't disclose a salary range for this listing. JobAssist shows pay whenever it's available.
This position can be done from anywhere, with no in-office requirement.
Yes – every application is tailored from your profile and this job's requirements, and you can review and edit before it's sent.