RL Training Framework Engineer - Scale ML Infrastructures

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

Benefits offered by this job

Equity

Medical insurance

Vision insurance

Dental insurance

401(k) retirement plan

Disability insurance

Life insurance

Job summary

SpaceXSI is seeking a Software Engineer to build and optimize the RL training framework, supporting everything from quick ablations to full production runs. You will design scales, profile performance, and ensure reliability across the RL stack.

Ideal candidates have experience with Python, Jax, Rust, or C++, and a background in distributed systems or large-scale ML infrastructure. SpaceXSI offers a comprehensive rewards package and equal opportunity employment.

Qualifications

• Experience with Python, Jax, Rust, and/or C++.
• Experience building and debugging large-scale distributed systems.
• Familiarity with RL training infrastructure.

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

Python

Jax

Rust

C++

Tools

Distributed systems

Job description

SpaceXSI is seeking a Software Engineer to build and optimize the RL training framework, supporting everything from quick ablations to full production runs. You will design scales, profile performance, and ensure reliability across the RL stack.

Ideal candidates have experience with Python, Jax, Rust, or C++, and a background in distributed systems or large-scale ML infrastructure. SpaceXSI offers a comprehensive rewards package and equal opportunity employment.