Member of Technical Staff - Mid-training

X AI
Palo Alto
Workplace: OnsiteFull timeUSD 180,000 - 440,000 annuallyFunction: Education & TrainingSkills: ["Communication","Prioritization","Problem-solving"]

The role focuses on building and scaling multimodal mid-training data pipelines for large AI models, including designing ML experiments, curating text/text+image+audio/video data, and optimizing data mixtures for RL. You’ll work with Spark, Ray, Python, and distributed systems in the Bay Area, contributing to the Foundation Model team and evaluating mid-training checkpoints.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
X AI
X AI
10 months ago

Member of Technical Staff - Mid-training

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 12 hours agoStatus: Live

Job Summary

The role focuses on building and scaling multimodal mid-training data pipelines for large AI models, including designing ML experiments, curating text/text+image+audio/video data, and optimizing data mixtures for RL. You’ll work with Spark, Ray, Python, and distributed systems in the Bay Area, contributing to the Foundation Model team and evaluating mid-training checkpoints.
Location: Palo Alto
Workplace: Onsite
Employment Type: Full time
Job Function: Education & Training
Seniority: Mid level

Key Responsibilities

  • •Scale synthetic coding data to trillions of tokens and verify at scale.
  • •Distill flagship model intelligence into flash models via synthetic data generation.
  • •Optimize mid-training data mixtures to improve RL ceilings.
  • •Engineer long-context data recipes for mid-training.
  • •Develop robust and diverse evaluation for mid-training checkpoints.

Pay and Benefits

Salary: USD 180,000 - 440,000 annually
Equity and Bonus:Equity
Perks:EquityHealth InsuranceVisionDental401kLife InsuranceDisability InsuranceDiscounts &

Key Requirements

  • •Expertise in ML and large model scaling; familiarity with scaling laws.
  • •Strong ability to design ML experiments and evaluate them.
  • •Experience curating AI training data across text, image, audio, and video modalities.
  • •Strong engineering abilities with Spark and Ray for large-scale data processing.
  • •Open to Bay Area relocation or proximity to Bay Area
Experience:Artificial intelligenceMachine learningResearch
Skills:CommunicationPrioritizationProblem-solving
Languages:English
Tech Stack:PythonJAXXLASparkRay

Company Brief

X AI
Develops advanced artificial intelligence models and research aimed at building safe, general AI and understanding the fundamental nature of the universe. Focuses on large-scale AI systems, research publications, and building foundational AI capabilities.
Industry: AI & Machine Learning
Company Size: Medium (51 to 250 employees)
Growth: Early Stage Startup
Headquarters: San Francisco, United States
Founded: 2023
Website