Founding AI Engineer

AimHire
5 days ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Languages
English
Compensation
$ 250K

Job location

Tech stack

Artificial Intelligence
Reinforcement Learning
PyTorch
Information Technology

Job description

We are looking for our first AI hire to own the research and implementation of our core agentic models. You will be responsible for turning groundbreaking research papers into robust, scalable systems. This is a rare opportunity to work on cutting-edge problems like long-horizon planning, tool-use optimization, and self-improving agents, with the compute resources to test your ideas at scale.

Requirements

  • Currently pursuing or recently completed a PhD/MSc in Computer Science, AI, or a related field.
  • Publication record at top conferences (NeurIPS, ICML, ICLR, ACL) is a huge plus.
  • Deep, hands-on experience with PyTorch, transformer architectures, and reinforcement learning.
  • Experience with frameworks like LangChain, LangGraph, or AutoGen is preferred, but a strong fundamental understanding of what they abstract is more important.
  • You read ML papers and immediately think about the implementation details and edge cases.
  • You are passionate about both theoretical rigor and shipping code that works.

About the company

We are a seed-stage startup, backed by Khosla Ventures and leading AI researchers, building the next generation of reasoning systems for AI agents. Our mission is to move beyond simple RAG and chain-of-thought, creating models that can dynamically plan, execute, and learn in complex environments. Our technical founder is a former Research Lead from Google DeepMind.

Apply for this position