Member Of Technical Staff, Mid-training (AI)

Location: Must be based in or open to relocating to the Bay Area (San Francisco and Palo Alto)

Company

xAI is focused on creating AI systems that enhance human understanding of the universe.

What you will do

• Design synthetic data algorithms to enhance model intelligence.
• Optimize mid-training data mixtures for reinforcement learning.
• Engineer long-context data recipes.
• Develop diverse evaluations for mid-training checkpoints.

Requirements

• Expertise in ML and large model scaling.
• Strong ability to design ML experiments.
• Familiarity with state-of-the-art AI training data techniques.
• Strong engineering skills in Spark, Ray, and large-scale data processing frameworks.

Benefits

• Equity and comprehensive medical, vision, and dental coverage.
• Access to a 401(k) retirement plan.
• Short & long-term disability insurance and life insurance.