Member of Technical Staff, Pre-training Scaling, Data and Eval jobs in United States
cer-icon
Apply on Employer Site
company-logo

xAI · 3 months ago

Member of Technical Staff, Pre-training Scaling, Data and Eval

xAI is focused on creating AI systems that can accurately understand the universe and aid humanity. The role involves working on multimodal pre-training to develop an omni model capable of processing text, image, video, and audio, with a focus on training large neural networks at scale and improving data quality across modalities.

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

Responsibilities

Training trillion parameter neural networks at scale, as well as a variety of smaller specialized models
Rapidly implementing the latest state-of-the-art methods from the deep learning literature
Innovating new ideas for pretraining and new scaling paradigm
Improving pretraining data quality at scale across different modalities

Qualification

PythonMLLarge model scalingDistributed trainingJAXXLARust / C++SparkPrioritization skillsCommunication skillsWork ethic

Required

Strong engineering skills with passion to improve different aspects of data and model
Expert in ML and large model scaling, familiar with all kinds of scaling laws
Familiar with distributed training, multi-GPU neural network training and experience on optimizing ML training efficiency
Familiar with state-of-the-art techniques for preparing AI training data
Good at organizing and meticulously bookkeeping data across multiple clouds, of multiple modalities, and from many sources
Strong communication skills
Ability to concisely and accurately share knowledge with teammates

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.

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
Greg Yang
Co-Founder
linkedin
leader-logo
Yuhuai Wu
Co-Founder
linkedin
Company data provided by crunchbase