Post-Training Researcher/Engineer
Description For Post-Training Researcher/Engineer
xAI is at the forefront of developing advanced AI systems that aim to understand the universe, create new knowledge, and achieve self-improvement through interactions. As a Post-Training Researcher/Engineer, you'll join the post-training team that transforms powerful pre-trained models into steerable, versatile systems capable of addressing real-world challenges.
The role combines cutting-edge research with practical engineering, focusing on enhancing model instruction-following capabilities and general usefulness. You'll work on creating research agendas, improving data mixtures, developing novel reinforcement learning algorithms, and implementing large-scale training frameworks.
The ideal candidate brings expert-level knowledge in machine learning, particularly in fine-tuning large language models, with a proven track record of impactful AI research. Experience with complex model optimizations, distributed systems, and advanced reinforcement learning is crucial.
Based in the Bay Area, you'll be part of a technical team that values direct interaction and rapid progress. The interview process is thorough yet efficient, designed to evaluate both technical skills and research capabilities, with a focus on exceptional work and vision alignment.
The role offers competitive compensation ranging from $180,000 to $440,000 USD, reflecting the high-impact nature of the position. You'll work with cutting-edge technologies including Python, Jax, and Rust, contributing to xAI's mission of advancing AI capabilities for real-world applications.
Responsibilities For Post-Training Researcher/Engineer
- Create and drive research agenda to advance model quality
- Improve data mixtures by building data collection pipelines and developing data generation techniques
- Create generalizable reward models and develop novel reinforcement learning algorithms
- Design and implement robust model evaluations
- Design and implement large-scale model training frameworks
- Collaborate with pre-training, reasoning, data, multimodal, applied, product efforts
Requirements For Post-Training Researcher/Engineer
- Expert in ML and fine-tuning large language models
- Track record in leading research 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
Jobs Related To xAI Post-Training Researcher/Engineer
