# Member of Technical Staff

**Company:** [xAI](https://hotfix.jobs/companies/xai)
**Location:** Palo Alto, CA
**Role:** ML Engineering
**Salary:** $180k – $440k/yr
**Experience:** 5+ years
**Skills:** Python, JAX, Rust, C++, Distributed Systems, Llm Training, Reinforcement Learning
**Posted:** 2026-07-14

> Build and optimize the RL training framework and infrastructure for large-scale workloads at SpaceXAI, from ablations to production runs. Requires experience with distributed systems and proficiency in Python, JAX, Rust, or C++.

## Job Description

## Responsibilities
- Design and implement the systems backing all RL workloads at SpaceXAI, 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 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++.

## Preferred Skills and Experience
- Experience with large scale LLM training infrastructure.
- Strong knowledge of reinforcement learning techniques.
- Experience with RL numerics.

## Compensation and Benefits
- $180,000 - $440,000 USD total compensation (base salary is one part of total rewards, which also includes equity and benefits).

## Similar roles

- [Member of Technical Staff, Applied Research](https://hotfix.jobs/jobs/d19460df-bf56-4f87-aa95-5e7bdf40798f) - LlamaIndex - San Francisco, CA - $180k – $250k/yr
- [Member of Technical Staff, Inference](https://hotfix.jobs/jobs/70280e8d-9e99-4be1-a488-b13edc2d4080) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Member of Technical Staff](https://hotfix.jobs/jobs/544a5b36-3a56-45fa-89fe-0873120ece22) - Black Forest Labs - San Francisco, CA - $180k – $290k/yr
- [Member of Technical Staff - X Search](https://hotfix.jobs/jobs/2480c779-d772-4053-b432-64b3e48d4cd3) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Member of Technical Staff - Post-Training and RL](https://hotfix.jobs/jobs/ed7e3533-4b73-4672-b792-579d9520ae65) - xAI - Palo Alto, CA - $180k – $600k/yr

**Apply:** https://hotfix.jobs/jobs/dd7c8bc4-741a-45d5-91a8-e6b1c4ad67e0
**Canonical:** https://hotfix.jobs/jobs/dd7c8bc4-741a-45d5-91a8-e6b1c4ad67e0