Member Of Technical Staff, Mid-training (AI)

TL;DR

Member Of Technical Staff, Mid-training (AI): Developing an omni model that comprehensively understands the universe through various modalities with an accent on synthetic data algorithms and mid-training data optimization. Focus on designing ML experiments and engineering robust evaluation methods for large-scale data processing.

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.