xAI

Member of Technical Staff - Mid-training

xAI
Apply
5 months ago
Palo Alto, CA, USASenior
H1B sponsor

Base Salary

$180k - $440k/yr

Responsibilities

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

Requirements

  • Expertise in machine learning and large-model scaling, including familiarity with scaling laws.
  • Strong ability to design machine-learning experiments.
  • Familiarity with state-of-the-art techniques for curating AI training data across text, image, audio, and video modalities.
  • Strong engineering ability with Spark, Ray, and other frameworks for large-scale data processing.

Benefits

  • $180,000–$440,000 base salary
  • Equity
  • Medical, vision, and dental coverage
  • 401(k) retirement plan
  • Short- and long-term disability insurance
  • Life insurance
  • Various discounts and perks

Categories

AI ResearchML Engineering
xAI

About xAI

1,001-5,000 employees

Understand the Universe. We are a team of AI technologists and business leaders on a mission to build AI systems that can help humanity understand the world better. https://x.ai/careers

Contact me