Reasoning Engineer

xAI is on a mission to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. As a Reasoning Engineer, you'll be at the forefront of building frameworks to enhance reasoning capabilities and distributed reinforcement learning systems.

The role involves working in a flat organizational structure where all employees are hands-on contributors. You'll be developing frameworks for reasoning improvement, building distributed RL systems, implementing techniques for inference time compute (including tree search and planning), and creating environments for agents.

The team is small, highly motivated, and focused on engineering excellence. This position is perfect for individuals who thrive on curiosity and enjoy challenging themselves. You'll be working with cutting-edge technologies including Python, JAX, and Rust, and will be expected to take ownership of components across the entire stack.

Key responsibilities include building robust and scalable distributed RL systems, optimizing frameworks for complex inference-time reasoning, and developing environments for agents. The ideal candidate should have experience with large-scale reinforcement learning systems, distributed systems design and implementation, and stay current with state-of-the-art RL algorithms.

The position offers a competitive salary range of $180,000 - $440,000 USD and is based in the Bay Area. The company values strong communication skills and the ability to share knowledge effectively with teammates. Leadership opportunities are available for those who show initiative and consistently deliver excellence.

The interview process is thorough but efficient, typically completed within a week, including technical assessments, systems hands-on evaluation, and project presentations. Join xAI to be part of a team that's pushing the boundaries of AI technology and contributing to humanity's understanding of the universe.