Model Engineer, RL Training Framework

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

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

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.