AI Engineering Lead

About the role:

Come define the future of AI Engineering at the company building the world’s leading agentic data science and analytics platform. As our AI Engineering Lead, you will accelerate the product and company by leveraging frontier AI technology and techniques, collaborating closely with product and engineering, and building and developing a world-class team of AI Engineers.

What you will do:

• Stay up-to-date on the latest in AI technology and work with engineering teams to push the frontier of product capabilities


• Define and implement the future of the AI Engineering function at Hex as we scale


• Hire and mentor an exceptional team of AI Engineers


• Support an exceptional developer and research experience for teams working with AI technology


• Deliver high-performing shared AI infrastructure, including search systems

About You:

• You have 4+ years of people management experience and leading high-performing teams


• You have deep familiarity with modern AI technology as well as classical ML systems and a passion for sharing that with others. Technologies of particular interest include LLMs and transformer models, agentic workflows, context management, and agentic evals.


• You are obsessed with driving business outcomes and delivering value to customers


• You are rigorous but pragmatic, balancing good experimental design with early-stage agility


• You are a strong communicator and love collaborating with other roles and teams to drive outcomes


• You are a talent magnet, with the ability to attract and retain exceptional AI talent

Our stack

Our product is a web-based notebook and app authoring platform. Our frontend is built with Typescript and React, using a combination of Apollo GraphQL and Redux for managing application state and data. On the backend, we also use Typescript to power an Express/Apollo GraphQL server that interacts with Postgres, Redis, and Kubernetes to manage our database and Python kernels. Our backend is tightly integrated with our infrastructure and CI/CD, where we use a combination of Terraform, Helm, and AWS to deploy and maintain our stack.

This is a full-time role based out of either our San Francisco or New York office. Employees are encouraged to come into the office at least twice a week, with Tuesdays and Thursdays being our main in-person days.

In addition to our unique culture, Hex proudly offers a competitive total rewards package, including but not limited to, market-benched salary & equity, comprehensive health benefits, and flexible paid time off.

The salary range for this role is: $246,000 - $329,000

The salary range shown may be a reflection of additional factors such as geographical location and skill ranges/levels we’re open to. Placement in the salary range will be decided upon completion of the interview process, taking into account factors like leaving room for growth, internal fairness & parity, your demonstrated skills, and the depth of your experience. Our Recruiting team will be able to provide more details during the interview process.