companyxAI logo

Technical Staff Member - Reasoning

xAILondon, UK
On-site Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Experience

Qualifications

We are seeking candidates with a strong coding background and experience in reinforcement learning or related fields. Familiarity with distributed systems, particularly in Python, JAX, or Rust, will be highly advantageous. A proactive approach to problem-solving and the ability to work collaboratively in a dynamic team environment are essential.

About the job

About xAI

At xAI, we are on a mission to develop advanced AI systems that deeply comprehend the intricacies of the universe while assisting humanity in its quest for knowledge. Our team is compact, highly driven, and dedicated to engineering excellence. We are looking for individuals who embrace challenges and thrive on curiosity. Operating with a flat organizational structure, we encourage all employees to be proactive and contribute directly to our mission. Leadership is awarded to those who demonstrate initiative and consistently achieve high standards. A strong work ethic and effective prioritization skills are essential. Excellent communication skills are expected, enabling team members to share knowledge clearly and concisely.

About the Position

As a Technical Staff Member, you will be responsible for developing frameworks to enhance reasoning capabilities, constructing distributed reinforcement learning systems, and implementing inference time computation techniques (such as tree search and planning). You will also create environments for agents, gaining exposure to a wide range of components across the entire technical stack.

Tech Stack

  • Python
  • JAX
  • Rust

Focus Areas

  • Develop robust and scalable distributed reinforcement learning systems.
  • Optimize frameworks for complex inference-time reasoning tasks.

About xAI

xAI is committed to pushing the boundaries of AI technology to help humanity better understand the universe. Our innovative team is dedicated to creating state-of-the-art AI systems that empower knowledge seekers.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.