2 months ago
Base Salary
$180k - $440k/yr
Responsibilities
- Design and implement systems supporting reinforcement-learning workloads from small-scale ablations through production training runs.
- Profile, debug, and optimize end-to-end training performance.
- Improve the scalability and observability of the reinforcement-learning stack.
Requirements
- Experience building, debugging, and optimizing the efficiency of large-scale distributed systems.
- Ability to work in unfamiliar areas and solve problems across all levels of the stack.
- Proficiency in Python, Jax, Rust, and/or C++.
- Experience with large-scale LLM training infrastructure is preferred.
- Strong knowledge of reinforcement-learning techniques is preferred.
- Experience with RL numerics is preferred.
Benefits
- Equity compensation.
- Comprehensive medical, vision, and dental coverage.
- 401(k) retirement plan.
- Short- and long-term disability insurance.
- Life insurance and additional discounts and perks.
Categories
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