Member of Technical Staff, Frontiers of Deep Learning Scaling jobs in United States
cer-icon
Apply on Employer Site
company-logo

xAI · 2 weeks ago

Member of Technical Staff, Frontiers of Deep Learning Scaling

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

xAI is focused on creating AI systems that enhance humanity's understanding of the universe. The role involves exploring new scaling paradigms for deep learning and addressing the challenges of large-scale training while ensuring stability and effectiveness in compute usage.

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

Responsibilities

Explore and answer what to scale up in terms of effective compute and useful data
Investigate new learning paradigms and model architectures for deep learning
Explore how to scale up large-scale and long-time training effectively

Qualification

PythonJAXPyTorchRustExecutionCommunication skillsSelf-drivenCollaboration

Required

Strong communication skills
Ability to concisely and accurately share knowledge with teammates
Experience with Python
Experience with JAX
Experience with PyTorch
Experience with Rust
Self-driven, smart and trustworthy

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
Palo Alto, 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
Yuhuai Wu
Co-Founder
linkedin
leader-logo
Toby Pohlen
Founding Member
linkedin

Recent News

Business Insider
xAI cofounder is taking a step back to 'go founder mode on my health' after Lyme disease diagnosis
2026-01-21
TechSpot
EPA shuts down xAI's off-grid turbine loophole at Colossus data center in Memphis
2026-01-20
9to5mac
Threads more popular than X on iPhone; Grok CSAM controversy intensifies
2026-01-20
Company data provided by crunchbase