AI Engineering Lead

Mosaic.tech; San Francisco (CA); On-site; USD 246,000 - 329,000; Full time

Benefits offered by this job

Market salary & equity

Comprehensive health benefits

Flexible paid time off

Job summary

Hex Technologies is seeking an AI Engineering Lead to shape the future of our AI engineering function. You will define strategy, hire and mentor a world-class AI team, and collaborate with product and engineering to accelerate our platform’s capabilities.

The role is based in SF or NY with in-office days recommended. The ideal candidate brings 4+ years of leadership experience, deep AI knowledge (LLMs, transformer models, agentic workflows), and a focus on delivering business value and customer

Qualifications

• 4+ years of people management experience leading high-performing teams.
• Deep familiarity with modern AI technology, LLMs, and transformer models.
• Interest in agentic workflows, context management, and agentic evals.
• Strong communicator with ability to collaborate across roles and teams.
• Talent magnet capable of attracting AI talent and delivering value.

Responsibilities

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

Skills

People management

AI technology

LLMs

Transformer models

Agentic workflows

Context management

Agentic evals

Communication

Talent magnet

Leadership

Tools

Typescript

React

Apollo GraphQL

Redux

Express

PostgreSQL

Redis

Kubernetes

Terraform

AWS

Helm

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‑bench 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.

Hex Technologies uses AI‑assisted tools as part of our application review process, including for resume screening and fraud detection. These tools help our team evaluate applications and verify applicant information. All AI-generated recommendations are reviewed by a member of our recruiting team before any hiring decision is made. No application is automatically rejected based solely on an AI tool's output.