xAI · Foundation models
Model Engineer, RL Training Framework
- Palo Alto
Still open on xAI's site · checked just now
- Posted
What you'll do
As written by xAI · Read the original
MODEL ENGINEER, RL TRAINING FRAMEWORK
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 SpaceXSI, 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
Requirements
BASIC QUALIFICATIONS
- Experience writing production-quality code in Python, PyTorch, JAX, Rust, or C++
PREFERRED SKILLS AND EXPERIENCE
- Experience building, debugging, and optimizing efficiency of large-scale distributed systems
- Experience with large scale LLM training infrastructure
- Strong knowledge of reinforcement learning techniques
- Experience with RL numerics
- Comfortable diving into unfamiliar areas and solving problems at all levels of the stack
Skills
Stack
- Python
- C++
- Rust
- PyTorch
- JAX
Working languages
English
Salary & benefits
COMPENSATION AND BENEFITS
$180,000 - $440,000 USD
Base salary is just one part of our total rewards package at SpaceXSI, 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.
SpaceXSI is an equal opportunity employer. For details on data processing, view our Recruitment Privacy Notice.
Company
Last checked . The details above were extracted automatically; the employer's original posting is the authority.
