Member of Technical Staff - Reasoning

ABOUT THE ROLE:

As a Member of Technical Staff, you will build frameworks to improve the reasoning capability, build distributed reinforcement learning systems, techniques for inference time compute (e.g. tree search and planning), and develop environments for agents.

You will get exposure and will be expected to solve and take ownership of components across the entire stack.

RESPONSIBILITIES:

• Build robust and scalable distributed RL systems.
• Optimise frameworks to enable complex inference-time reasoning.
• Develop environments and harnesses for agents.

BASIC QUALIFICATIONS:

• Experienced with large-scale reinforcement learning systems.
• Designing and implementing distributed systems.
• Keeping up with state-of-the-art RL and inference time compute algorithms.

COMPENSATION AND BENEFITS:

£107,000 - £262,000 GBP

Base salary is just one part of our total rewards package at xAI, which also includes equity, comprehensive healthcare and dental coverage, cash plan, income protection, life insurance, and various other discounts and perks.