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

xAI · 4 hours ago

Member of Technical Staff, Pre-training Data Scaling

xAI is focused on creating AI systems that enhance humanity's pursuit of knowledge. The role involves collaborating with teams to source datasets, architecting data transformation pipelines, and innovating methods for scaling pre-training data for model training.

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

Responsibilities

Collaborate with the crawling team to discover and source datasets
Architect pipelines to transform datasets at petabyte scales with efficiency and precision
Craft insightful experiments to assess dataset quality and impact on model training
Design careful control experiments to understand the data and data pipeline
Innovate the recipe for scaling pre-training data to new frontiers

Qualification

PythonSparkRayRustML expertiseFirst-principles thinkingEngineering abilitiesCommunication skills

Required

Strong communication skills
Ability to concisely and accurately share knowledge with teammates
Hands-on contribution to the company's mission
Strong work ethic and prioritization skills

Preferred

First-principles thinking — the ability to think deeply and question every assumption rather than accepting the status quo
Expertise in ML and large model scaling, with familiarity across all kinds of scaling laws
Strong ability to design ML experiments
Knowledge of state-of-the-art techniques for curating pre-training data
Strong engineering abilities in Spark, Ray, and other frameworks for large-scale data processing

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
Growth Stage
Total Funding
$22.73B
Key Investors
Neptune Digital AssetsSpaceXMorgan Stanley
2025-12-11Secondary Market· $0.3M
2025-07-13Corporate Round· $5.32B
2025-07-01Debt Financing· $5B

Leadership Team

leader-logo
Toby Pohlen
Founding Member
linkedin
Company data provided by crunchbase