# Member of Technical Staff, Mid-training

**Company:** [xAI](https://hotfix.jobs/companies/xai)
**Location:** Palo Alto, CA, San Francisco, CA
**Role:** ML Engineering
**Salary:** $180k – $440k/yr
**Skills:** Python, JAX, Xla, Spark, Ray, Machine Learning, LLMs, Scaling Laws, Data Processing, Synthetic Data
**Posted:** 2026-03-06

> Develops multimodal mid-training data pipelines for omni models handling text, image, video, and audio. Requires ML expertise, scaling laws knowledge, experiment design, and strong engineering in large-scale data frameworks like Spark and Ray.

## Job Description

## Tech Stack
- Python
- JAX and XLA
- Spark
- Ray

## Focus
- Scale synthetic coding data to trillions of tokens with large-scale docker verification.
- Distill the intelligence of flagship models into flash models through synthetic data generation.
- Optimize mid-training data mixtures to boost the ceiling for RL.
- Engineer long-context data recipes.
- Develop robust and diverse evaluation for mid-training checkpoints.

## Ideal Experience
- Expertise in ML and large model scaling, with familiarity across all kinds of scaling laws.
- Strong ability to design ML experiments.
- Familiarity with state-of-the-art techniques for curating AI training data for text, image, audio, and video modalities.
- Strong engineering abilities in Spark, Ray, and other frameworks for large-scale data processing.

## Annual Salary Range
$180,000 - $440,000 USD

## Benefits
Base salary is just one part of our total rewards package at xAI, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks.

## Similar roles

- [Member of Technical Staff](https://hotfix.jobs/jobs/dd7c8bc4-741a-45d5-91a8-e6b1c4ad67e0) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Member of Technical Staff, Applied Research](https://hotfix.jobs/jobs/d19460df-bf56-4f87-aa95-5e7bdf40798f) - LlamaIndex - San Francisco, CA - $180k – $250k/yr
- [Member of Technical Staff, Inference](https://hotfix.jobs/jobs/70280e8d-9e99-4be1-a488-b13edc2d4080) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Member of Technical Staff](https://hotfix.jobs/jobs/544a5b36-3a56-45fa-89fe-0873120ece22) - Black Forest Labs - San Francisco, CA - $180k – $290k/yr
- [Member of Technical Staff - X Search](https://hotfix.jobs/jobs/2480c779-d772-4053-b432-64b3e48d4cd3) - xAI - Palo Alto, CA - $180k – $440k/yr

**Apply:** https://hotfix.jobs/jobs/bd5c8b59-aa38-43e8-a44d-c0ae61fafbb3
**Canonical:** https://hotfix.jobs/jobs/bd5c8b59-aa38-43e8-a44d-c0ae61fafbb3