xAI
Member of Technical Staff - RL Training Framework
Palo Alto, CA · Staff+
Sponsorship not specified$180k-$440kDetected 1 day ago
PythonC++Distributed SystemsLLMsLeadershipCommunication
About the role
- The RL infrastructure team is looking for an engineer to help develop our RL training framework.
Responsibilities
- 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
- Experience building, debugging, and optimizing efficiency of large-scale distributed systems
Requirements
- Proficiency in Python, Jax, Rust, and/or C++
Skills
- Experience with large scale LLM training infrastructure
- Strong knowledge of reinforcement learning techniques
- Experience with RL numerics
- This organization is for individuals who appreciate challenging themselves and thrive on curiosity.
- All employees are expected to be hands-on and to contribute directly to the company's mission.
- Work ethic and strong prioritization skills are important.
- All employees are expected to have strong communication skills.
- They should be able to concisely and accurately share knowledge with their teammates.
Compensation
- $180,000 - $440,000 USD
- Base salary is just one part of our total rewards package at SpaceXAI, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks.
Equal opportunity
- equal opportunity employer.
This listing is sourced directly from xAI's careers page and normalized into a canonical job model.