London AI Tech Lead: Distributed RL & Inference

xAI in London is seeking a Member of Technical Staff (AI) to design and implement robust distributed reinforcement learning systems and frameworks for inference-time reasoning. You will develop advanced environments and testing harnesses for AI agents, while owning components across the stack and collaborating with global teams to tackle concurrency and architectural challenges.

The role requires in-person attendance in London, strong expertise in Python, JAX, and Rust, and experience with