GrepJob
xAI

Member of Technical Staff - RL Training Framework

xAI
Apply
about 8 hours ago
Palo Alto, CA, USAMid Level / Senior
H1B Sponsor

Base Salary

$180k - $440k/yr

Responsibilities

  • Design and implement systems for RL workloads at SpaceXAI.
  • Profile, debug, and optimize end-to-end training performance.
  • Improve scalability and observability of the RL stack.

Requirements

  • Experience building, debugging, and optimizing large-scale distributed systems.
  • Comfortable solving problems at all levels of the stack.
  • Proficiency in Python, Jax, Rust, and/or C++.

Benefits

  • Equity in the company.
  • Comprehensive medical, vision, and dental coverage.
  • Access to a 401(k) retirement plan.
  • Short & long-term disability insurance.
  • Life insurance and various discounts and perks.

Tech Stack

Categories

AI & MLData Engineering