Senior RL Model Engineer: Large-Scale Training

xAI; Palo Alto (CA); On-site; USD 180,000 - 400,000; Full time

Benefits offered by this job

Equity

Medical, Vision, and Dental coverage

401(k) retirement plan

Disability insurance

Life insurance

Discounts and perks

Job summary

SpaceXSI in Palo Alto is seeking a Sr. Model Engineer for RL Training Framework to design and optimize the systems powering RL workloads, from ablations to production runs. You will work on scalable, observable RL stacks and large-model training infrastructure.

Responsibilities include implementing production-grade code in Python/C++/Rust, profiling performance, and improving efficiency across distributed systems, with equity and comprehensive benefits as part of SpaceXSI's total rewards.

Qualifications

• Production-quality code in Python, PyTorch, JAX, Rust, or C++.
• Experience training large models with 1B+ parameters.
• Experience with large-scale distributed systems and RL training infrastructure.

Responsibilities

• Design and implement the systems backing RL workloads from ablations to production.
• Profile, debug, and optimize end-to-end training performance.
• Improve scalability and observability of the RL stack.

Skills

Python

PyTorch

JAX

Rust

C++

Job description

SpaceXSI in Palo Alto is seeking a Sr. Model Engineer for RL Training Framework to design and optimize the systems powering RL workloads, from ablations to production runs. You will work on scalable, observable RL stacks and large-model training infrastructure.

Responsibilities include implementing production-grade code in Python/C++/Rust, profiling performance, and improving efficiency across distributed systems, with equity and comprehensive benefits as part of SpaceXSI's total rewards.