ML Systems Engineer — Large-Scale AI Training

xAI; Palo Alto (CA); Presencial; USD 180.000 - 440.000; Jornada completa

Ventajas ofrecidas por este puesto de trabajo

Equity

401(k) retirement plan

Medical, Vision, and Dental coverage

Disability insurance

Discounts and perks

Descripción de la vacante

SpaceXSI is building a comprehensive AI stack and pursuing massive-scale ML research in a collaborative environment. The role focuses on scaling data, designing experiments, and evaluating mid-training checkpoints to push model capabilities across text, image, audio, and video modalities.

Based in Palo Alto, this position offers a competitive compensation package, equity, and extensive health coverage as part of the total rewards.

Formación

• Experience with ML and large model scaling.
• Familiarity with large-scale data processing frameworks.

Responsabilidades

• 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.

Conocimientos

ML model scaling

Experiment design

Data processing

RL workflows

Herramientas

Spark

Ray

Docker

Descripción del empleo

SpaceXSI is building a comprehensive AI stack and pursuing massive-scale ML research in a collaborative environment. The role focuses on scaling data, designing experiments, and evaluating mid-training checkpoints to push model capabilities across text, image, audio, and video modalities.

Based in Palo Alto, this position offers a competitive compensation package, equity, and extensive health coverage as part of the total rewards.