AI Engineering Lead

Hex; California (MO); Hybrid; USD 246,000 - 329,000; Full time

Benefits offered by this job

Total rewards package

Health benefits

Flexible time off

Job summary

Hex is seeking an AI Engineering Lead to define and scale the AI Engineering function while building a world-class team of AI Engineers. You will hire, mentor, and collaborate with product and engineering to accelerate the platform’s capabilities using LLMs and transformer models.

You will lead efforts to deliver robust AI infrastructure and excellent developer experiences, shaping the company’s AI strategy and technical direction across teams.

Qualifications

• 4+ years of people management experience leading high-performing teams.
• Deep familiarity with modern AI technology including LLMs, transformer models, agentic workflows, context management, and agentic evals.
• Driven to deliver business value and customer outcomes.
• Rigorous yet pragmatic with a balance of experimentation and speed.
• Strong communicator and collaborator across roles and teams.

Responsibilities

• Stay up-to-date on the latest in AI technology and collaborate with engineering to push product capabilities.
• Define and implement the AI Engineering function as Hex scales.
• Hire and mentor an exceptional team of AI Engineers.
• Support a strong developer and research experience for teams working with AI.
• Deliver high-performing shared AI infrastructure, including search systems.

Skills

People management

AI leadership

Transformer models

LLMs

Communication

Tools

Kubernetes

Postgres

Redis

Terraform

AWS

Job description

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.

By submitting an application the candidate consents to the use of their personal information in accordance with the Hex Privacy policy:https://learn.hex.tech/docs/trust/privacy-policy.

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.

About Hex

Hex is growing our team of builders on a mission to make everyone a data person. Our platform solves key pain points with today’s data and analytics tooling, and empowers anyone to explore data using natural language, with or without code, on trusted context. Thousands of customers like Ramp, Figma, Stubhub, Anthropic, and Gamma love Hex for our beautiful UI, agentic superpowers, and boundless flexibility.

Hex has raised over $100M from leading investors like Sequoia, a16z, Snowflake and Amplify. With office hubs in San Francisco and New York, as well as remote team members throughout the US and UK, we’re a high-agency team that ships fast and has fun doing it. Check out our Intro to Hex video and employee handbook to learn more about what it’s like to work at Hex.