AI Engineering Lead

Posted Yesterday
Be an Early Applicant
2 Locations
Hybrid
246K-329K Annually
Mid level
Artificial Intelligence • Big Data • Software • Analytics • Business Intelligence • Big Data Analytics
Hex empowers thousands of people to ask & answer new questions of data, and seamlessly share the results with everyone.
The Role
Lead the AI Engineering function by advancing frontier AI capabilities, defining team strategy, hiring and mentoring AI engineers, supporting AI developer and research experiences, and delivering shared infrastructure such as search systems. The role collaborates closely with product and engineering teams and requires expertise in LLMs, transformer models, agentic workflows, context management, and AI evaluation.
Summary Generated by Built In
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.

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.

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

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

Skills Required

  • 4+ years of people management experience and experience leading high-performing teams
  • Deep familiarity with modern AI technology and classical machine learning systems
  • Knowledge of LLMs and transformer models
  • Knowledge of agentic workflows, context management, and agentic evaluations
  • Passion for sharing AI and machine learning expertise with others
  • Ability to drive business outcomes and deliver customer value
  • Rigorous and pragmatic approach to experimental design and early-stage execution
  • Strong communication and cross-functional collaboration skills
  • Ability to attract and retain exceptional AI talent

Hex Compensation & Benefits Highlights

  • Healthcare Strength — Official materials and benefits profiles consistently reference comprehensive medical, dental, and vision coverage, with mental-health support and FSAs also noted. Some listings additionally cite very high employer premium contributions on the base plan.
  • Parental & Family Support — Benefits information highlights fully paid parental leave (e.g., 12 weeks, or 16 for the gestational parent) with flexible return-to-work support. Family-planning benefits such as Carrot with a quoted lifetime allowance are also described.
  • Retirement Support — Multiple profiles note a 401(k) with a company match (commonly cited up to 4%). This retirement support is presented as part of a broader market-benchmarked total-rewards package.

Hex Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, CA
160 Employees
Year Founded: 2019

What We Do

Hex is changing the way people work with data. Our platform makes analytics workflows more powerful, collaborative, and shareable. Hex solves key pain points with today's data and analytics tooling, and is loved by thousands of users all over the world for the beautiful UI, new superpowers, and boundless flexibility. We are a tight-knit crew of engineers, designers, and data aficionados. Our roadmap is full of big ideas and little details, and we would love your help bringing them to life. Hex has raised over $100m from great VCs and angels, giving us many years of runway and the ability to pay competitive salaries, offer great benefits, and provide meaningful equity.

Why Work With Us

We’re a team of builders. We spend our days designing and developing beautiful products, growing and supporting our user base, and helping each other succeed and learn. To do that, we've developed a culture focused on quality, craft, and speed -- with very few meetings, bias toward action, and taking care of each other through feedback and support.

Gallery

Gallery
Gallery
Gallery
Gallery

Hex Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

We have a truly hybrid workforce, with many employees working fully remotely, and others coming into one of our two offices (SF or NYC) 2-3 days per week.

Typical time on-site: Flexible
HQSan Francisco, CA
New York, NY
Learn more

Similar Jobs

Hex Logo Hex

Mid-market Account Executive

Artificial Intelligence • Big Data • Software • Analytics • Business Intelligence • Big Data Analytics
Remote or Hybrid
2 Locations
160 Employees
250K-250K Annually

Hex Logo Hex

Product Manager

Artificial Intelligence • Big Data • Software • Analytics • Business Intelligence • Big Data Analytics
Remote or Hybrid
United States
160 Employees
177K-300K Annually

Hex Logo Hex

Account Manager

Artificial Intelligence • Big Data • Software • Analytics • Business Intelligence • Big Data Analytics
Hybrid
New York, NY, USA
160 Employees
200K-215K Annually

Hex Logo Hex

Solutions Engineer

Artificial Intelligence • Big Data • Software • Analytics • Business Intelligence • Big Data Analytics
Remote or Hybrid
United States
160 Employees
177K-237K Annually

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account