Job Description

xAI is seeking a Member of Technical Staff to join the post-training team, focusing on transforming pre-trained models into steerable, versatile models for real-world applications. The role involves advancing models for tool calling, reasoning, and model coordination, with an emphasis on interactive and agentic behaviors to enable seamless human-AI collaboration.

Responsibilities

• Enhance model capabilities in interactive tool calling, structured reasoning, and model orchestration
• Enhancing data mixtures through data collection pipelines
• Developing reward models and reinforcement learning algorithms
• Designing robust evaluations for real-world interactive dynamics
• Implementing large-scale training frameworks
• Collaborating with various teams to expand interactive AI frontiers

Requirements

• Expert in ML and training large language models
• Track record in leading studies impacting AI advancement
• Experience in data-driven large language model behavior improvements
• Experience in reinforcement learning or inference-time search techniques
• Experience in developing benchmarks or distributed machine learning systems
• Experience in model optimizations under complex setups
• Experience in model routing

Benefits

• No benefits