AI Engineer - Reasoning
180k - 440k USD
On-site
Full Time
#Engineering
#Python
#Jax
#Rust
#Reinforcement Learning
#Distributed Systems
You will join a focused team building next-generation AI systems that push the boundaries of machine reasoning. Our work centers on creating frameworks that let agents think more deeply, plan over longer horizons, and learn efficiently at massive scale. We operate in the Bay Area and value engineers who want to own substantial parts of the stack while staying close to the latest research in reinforcement learning and inference-time compute.
What you'll be doing
- Design and scale distributed reinforcement learning systems that train agents across many machines while maintaining stability and speed.
- Build and refine frameworks that improve reasoning quality through techniques such as tree search, planning, and other inference-time compute methods.
- Create environments and harnesses that let agents interact with complex tasks, then measure and improve their performance.
What you'll bring
- Hands-on experience designing and running large-scale reinforcement learning systems.
- Strong background in distributed systems, with the ability to implement and optimize them in practice.
- Familiarity with current state-of-the-art algorithms in reinforcement learning and inference-time compute, along with the drive to apply them in production settings.
- Proficiency in Python, JAX, and Rust.
- Fluency in English.
What you'll get
We offer a salary range of $180,000 - $440,000 USD. You will work on-site in the Bay Area with colleagues who review every application directly and keep the interview process to roughly one week. The process includes a short initial call followed by four technical interviews: a coding assessment in the language of your choice, a live systems problem-solving session, a project deep-dive where you present prior work, and a final meet-and-greet with the broader team. All interviews take place over Google Meet, and no recruiters are involved in assessments.
