Member of Technical Staff - Mid-training

2 weeks agoSalary: 180K - 440KPalo Alto, USAAiJobs by SpaceXAI

SpaceXAI builds frontier AI models and products including the Grok conversational assistant and a multimodal developer API.

Austin, United States
About SpaceXAI

SpaceXAI is a US-based AI company focused on accelerating scientific discovery and understanding the universe through reasoning, voice, image, video, and generative AI systems. Formerly xAI, it was acquired by SpaceX effective February 2, 2026, while continuing to operate the x.ai site and services.

View jobs by SpaceXAI

Skills

Candidate Availability

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You will scale synthetic coding data, distill flagship-model capabilities into flash models, optimize mid-training data mixtures, engineer long-context data recipes, and develop robust evaluations for model checkpoints.

Requirements

  • Apply machine-learning and large-model scaling expertise
  • Design machine-learning experiments
  • Curate AI training data for text, image, audio, and video
  • Use Spark, Ray, and large-scale data-processing frameworks

Responsibilities

  • Scale synthetic coding data with large-scale Docker verification
  • Distill flagship-model intelligence into flash models through synthetic data generation
  • Optimize mid-training data mixtures for reinforcement learning
  • Engineer long-context data recipes
  • Develop robust and diverse evaluations for mid-training checkpoints

Benefits

  • Equity
  • Medical coverage
  • Vision coverage
  • Dental coverage
  • 401(k) retirement plan
  • Short-term disability insurance
  • Long-term disability insurance
  • Life insurance
  • Discounts and perks
Member of Technical Staff - Mid-training at SpaceXAI | JobStash