Member of Technical Staff - Multimodal Interactions Post-training

xAI • East Palo Alto, CA

84d • $180,000 - $440,000

About The Position

The post-training team at xAI transforms powerful pre-trained models into steerable, versatile models capable of understanding and addressing real-world challenges. As a Member of Technical Staff on the post-training team, you'll advance models for tool calling, reasoning, and model coordination. You'll work across language models of different scales, modalities, and specializations to create novel interactive paradigms. With a strong emphasis on interactivity and agentic behaviors, you will pioneer a new class of orchestration models to unlock seamless, real-time, human-AI collaboration.

Requirements

• Expert in ML and training large language models.
• Track record in leading studies that significantly impacts AI advancement.
• Experience in data-driven large language model behavior improvements.
• Experience in advanced reinforcement learning or inference-time search techniques.
• Experience in developing benchmarks or large-scale distributed machine learning systems.
• Experience in model optimizations under complex setups (e.g., multi-modality, multi-context, multi-agent, long-horizon tasks, diverse user preference/feedback).
• Experience in model routing.

Responsibilities

• Enhance model capabilities in interactive tool calling, structured reasoning, and model orchestration, focusing on real-time human-AI interactions and agentic decision-making.
• Enhance data mixtures through data collection pipelines and generation techniques optimized for interactive human-AI collaborations and orchestration.
• Develop generalizable reward models and novel reinforcement learning algorithms to promote emergent interactive behaviors, enabling models to guide heavier backend systems.
• Design robust evaluations that capture real-world interactive dynamics, human-AI synergy, and orchestration across diverse backend models, scales, and specializations.
• Implement large-scale training frameworks for models that integrate and orchestrate backend models of varying sizes, skills, and capabilities in interactive, agentic environments.
• Collaborate with pre-training, reasoning, data, multimodal, applied, and product teams to expand interactive AI frontiers.

Benefits

• Equity
• Comprehensive medical coverage
• Vision coverage
• Dental coverage
• Access to a 401(k) retirement plan
• Short & long-term disability insurance
• Life insurance
• Various discounts and perks

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Number of Employees: 1,001-5,000 employees