RL Training Infrastructure Engineer

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

Benefits offered by this job

Equity

Medical insurance

Dental coverage

Vision coverage

401(k)

Disability insurance

Life insurance

Employee discounts

Job summary

SpaceXSI is seeking an engineer to help develop our RL training framework. The role focuses on designing and implementing the systems that back all RL workloads, from small-scale ablations to full production training runs.

You will profile, debug, and optimize end-to-end training performance while improving scalability and observability of the RL stack. Candidates should have experience with large-scale distributed systems and proficiency in Python, Jax, Rust, or C++.

Qualifications

• Experience building, debugging, and optimizing efficiency of large-scale distributed systems.
• Comfortable diving into unfamiliar areas and solving problems at all levels of the stack.
• 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

Large-scale distributed systems

Problem solving

Python

Jax

Rust

C++

Job description

SpaceXSI is seeking an engineer to help develop our RL training framework. The role focuses on designing and implementing the systems that back all RL workloads, from small-scale ablations to full production training runs.

You will profile, debug, and optimize end-to-end training performance while improving scalability and observability of the RL stack. Candidates should have experience with large-scale distributed systems and proficiency in Python, Jax, Rust, or C++.