Member of Technical Staff, RL Training Framework jobs in United States
cer-icon
Apply on Employer Site
company-logo

xAI · 4 months ago

Member of Technical Staff, RL Training Framework

positionPalo Alto, CA
timeFull-time
remoteOnsite
seniorityMid Level
money$180K/yr - $440K/yr

xAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. They are seeking a Member of Technical Staff to design and implement distributed RL systems and optimize system performance.

Artificial Intelligence (AI)Foundational AIGenerative AIInformation TechnologyMachine Learning
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Design and implement state-of-the-art distributed RL systems
Profile, debug, and optimize system performance
Software and algorithm co-design with engineers

Qualification

Distributed RL systemsLarge scale training systemsAsync RL training frameworksReinforcement learning techniquesPythonJaxRustPrioritization skillsCommunication skillsWork ethic

Required

Experience building, debugging, and optimizing large scale distributed training systems
Experience building async RL training frameworks
Experience in inference systems
Proficiency in Python, Jax, or Rust
Strong communication skills
Ability to concisely and accurately share knowledge with teammates
Strong work ethic and prioritization skills

Preferred

PhD with AI or other scientific domains
Strong knowledge of reinforcement learning techniques
Experience building infra for large-scale reinforcement learning and multi-agent reinforcement learning

Benefits

Equity
Comprehensive medical, vision, and dental coverage
Access to a 401(k) retirement plan
Short & long-term disability insurance
Life insurance
Various other discounts and perks

Company

xAI

twittertwittertwitter
company-logo
XAI is an artificial intelligence startup that develops AI solutions and tools to enhance reasoning and search capabilities.
dateFounded in 2023
location
Burlingame, California, USA
people-size11-50 employees
web-link
https://x.ai

H1B Sponsorship

xAI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1)

Funding

Current Stage
Late Stage
Total Funding
$42.73B
Key Investors
Neptune Digital AssetsSpaceXMorgan Stanley
2026-01-06Series E· $20B
2025-12-11Secondary Market· $0.3M
2025-07-13Corporate Round· $5.32B

Leadership Team

leader-logo
Toby Pohlen
Founding Member
linkedin

Recent News

Livemint.com
Germany plans measures to combat harmful AI image manipulation
2026-01-11
Clean Technica
Elon Musk & Grok Celebrate Their War On Women
2026-01-11
CBS News
U.K. says ban on Elon Musk's X platform "on the table" over Grok AI sexualized images
2026-01-11
Company data provided by crunchbase