Member of Technical Staff RL Inference
SpaceXAI builds frontier AI models and products including the Grok conversational assistant and a multimodal developer API.
Funding history
About SpaceXAI
SpaceXAI is a US-based AI company focused on accelerating scientific discovery and understanding the universe through reasoning, voice, image, video, and generative AI systems. Formerly xAI, it was acquired by SpaceX effective February 2, 2026, while continuing to operate the x.ai site and services.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will design and optimize inference systems for reinforcement-learning workloads, from small-scale ablations through production training runs. You will analyze performance bottlenecks and work with modeling specialists to implement new reinforcement-learning techniques and algorithms efficiently.
Requirements
- Experience building, debugging, and optimizing large-scale distributed systems.
- Experience in LLM inference.
- Proficiency in Python, C++, or Rust.
- Knowledge of PyTorch, JAX, or CUDA.
Responsibilities
- Design and optimize the inference stack for reinforcement-learning workloads.
- Analyze, profile, and address performance bottlenecks in large-scale reinforcement-learning systems.
- Implement novel reinforcement-learning techniques and algorithms efficiently.
Benefits
- Equity
- Medical coverage
- Vision coverage
- Dental coverage
- 401(k) retirement plan
- Short-term disability insurance
- Long-term disability insurance
- Life insurance
- Discounts and perks