AI Infrastructure Engineer

xAI is at the forefront of transforming pre-trained AI models into versatile, steerable systems capable of tackling real-world challenges. As an AI Infrastructure Engineer, you'll join the post-training team to develop and optimize frameworks for large-scale machine learning tasks, with a focus on reinforcement learning and agent systems.

The role demands expertise in building high-performance, scalable software that pushes the boundaries of AI capabilities. You'll be responsible for creating efficient frameworks for model fine-tuning, reinforcement learning, and large-scale agent simulation. Your work will directly impact the development of synthetic data generation and model-based improvements.

The ideal candidate brings deep expertise in distributed machine learning systems, GPU optimization, and modern ML frameworks like JAX or PyTorch. You'll work with cutting-edge technologies including Python, Rust, and CUDA, while maintaining high standards for code quality and testing.

Based in the vibrant Bay Area tech hub, you'll join a team committed to advancing AI technology. The comprehensive interview process reflects our technical focus, with coding assessments and infrastructure design challenges evaluated directly by our technical team. With a competitive salary range of $180,000 - $440,000 USD, this role offers an opportunity to shape the future of AI technology while working with state-of-the-art systems and frameworks.

The position offers a unique chance to work on breakthrough AI technologies while collaborating with a team of exceptional engineers and researchers. Your contributions will help define the future of AI systems and their applications in solving real-world problems.