Senior RL Training Engineer – Scalable AI Stack

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

Benefits offered by this job

Equity

Medical coverage

Vision coverage

Dental coverage

401(k) plan

Disability insurance

Life insurance

Employee discounts

Job summary

SpaceXSI in Palo Alto seeks a Senior Model Engineer to help build our RL training framework. You will design and implement systems for RL workloads, from small-scale ablations to production runs, and collaborate across teams to push the limits of large-scale AI.

You should have hands-on experience with Python, PyTorch, JAX, Rust, or C++, plus a track record of training models with 1B+ parameters. The role rewards innovation with a competitive salary, equity, and comprehensive benefits.

Qualifications

• Experience writing production-quality code in Python, PyTorch, JAX, Rust or C++.
• Experience training large models (1B+ parameters) including RL training.

Responsibilities

• Design and implement the systems backing all RL workloads, from ablations to production training.
• 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 seeks a Senior Model Engineer to help build our RL training framework. You will design and implement systems for RL workloads, from small-scale ablations to production runs, and collaborate across teams to push the limits of large-scale AI.

You should have hands-on experience with Python, PyTorch, JAX, Rust, or C++, plus a track record of training models with 1B+ parameters. The role rewards innovation with a competitive salary, equity, and comprehensive benefits.