29 days ago
Palo Alto, CA, USAMid Level / Senior
H1B Sponsor
Base Salary
$180k - $440k/yr
Responsibilities
- Design and implement systems for RL workloads at SpaceXAI.
- Profile, debug, and optimize end-to-end training performance.
- Improve scalability and observability of the RL stack.
Requirements
- Experience building, debugging, and optimizing large-scale distributed systems.
- Comfortable solving problems at all levels of the stack.
- Proficiency in Python, Jax, Rust, and/or C++.
Benefits
- Equity in the company.
- Comprehensive medical, vision, and dental coverage.
- Access to a 401(k) retirement plan.
- Short & long-term disability insurance.
- Life insurance and various discounts and perks.
About xAI
Understand the Universe. We are a team of AI technologists and business leaders on a mission to build AI systems that can help humanity understand the world better. https://x.ai/careers