Senior RL Training Framework Engineer

SpaceXSI; Palo Alto (CA); On-site; USD 180,000 - 440,000; Full time

Benefits offered by this job

Medical, Vision & Dental

401(k) retirement plan

Equity

Disability & Life Insurance

Job summary

SpaceXSI is seeking an engineer to help develop our RL training framework within the RL infrastructure team. You will design and implement systems underpinning all RL workloads, from ablations to full-scale production runs, and work to profile, debug, and optimize training performance.

You will also enhance scalability and observability across the RL stack. The role emphasizes building robust, distributed systems and requires comfort with cross-stack challenges.

Qualifications

• Experience building, debugging and optimizing large-scale distributed systems.
• Ability to tackle problems across the entire stack and learn new areas quickly.
• Proficiency in Python, Jax, Rust and/or C++.

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.

Skills

Distributed systems

Python

C++

Rust

Jax

Job description

SpaceXSI is seeking an engineer to help develop our RL training framework within the RL infrastructure team. You will design and implement systems underpinning all RL workloads, from ablations to full-scale production runs, and work to profile, debug, and optimize training performance.

You will also enhance scalability and observability across the RL stack. The role emphasizes building robust, distributed systems and requires comfort with cross-stack challenges.