# AI Research Engineer

**Company:** [Hex](https://hotfix.jobs/companies/hex)
**Location:** San Francisco, CA, New York, NY
**Role:** ML Engineering
**Salary:** $214k – $285k/yr
**Experience:** 5+ years
**Skills:** Python, TypeScript, React, Kubernetes, MLOps, GraphQL, Postgres, Redis, Terraform, AWS
**Posted:** 2026-04-09

> Builds and deploys production AI features like Notebook Agent for data science workflows, partnering with product teams on experiments, model fine-tuning, and infra. Requires senior AI/ML engineering experience with MLOps, Python/TS proficiency.

## Job Description

## Responsibilities
- Partner with product teams to build AI experiences like the Notebook Agent.
- Run experiments, fine-tune models, deploy AI infrastructure, and build/maintain experimentation tooling.
- Drive Hex's context engine and advance Notebook Agent capabilities (SQL/Python generation, reports, visualizations).
- Build features from 0 to 1.
- Determine architecture and stack for AI-enabled capabilities.
- Ship product experiences changing data science/analyst workflows.

## Requirements
- Senior engineer from AI Eng, SWE, or MLE background.
- Experience getting AI/ML capabilities into production for real users.
- Enthusiasm for AI applications to business problems.
- Understanding of core MLOps/SW Architecture for modern ML apps (strong on Infra/MLOps ideal).
- Comfortable in Python & JS/TS.
- Experimentalist mindset with quick iteration and critical thinking.
- Interest in data space, shipping products, empowering users.
- High quality bar for design, correctness, testing.

## Our Stack
- Frontend: **Typescript**, **React**, **Apollo GraphQL**, **Redux**.
- Backend: **Typescript**, **Express**, **Apollo GraphQL**, **Postgres**, **Redis**, **Kubernetes**.
- Infra/CI/CD: **Terraform**, **Helm**, **AWS**.

## Compensation
- Salary range: **$214,000 - $285,000** (USD), varies by location, skills, experience.

## Similar roles

- [Machine Learning Engineer](https://hotfix.jobs/jobs/97713038-58cb-46c5-a9c7-0e1554e1ba4f) - Liftoff - Remote - $215k – $275k/yr
- [Software Engineer, Enterprise AI](https://hotfix.jobs/jobs/98385fe7-6f4c-4386-ab6e-d4033a6d2440) - Scale AI - New York, NY - $216k – $270k/yr
- [Machine Learning Engineer - Embedded Insights](https://hotfix.jobs/jobs/aff6646e-3664-4f34-a9db-69fd66c3eca7) - Plaid - San Francisco, CA - $212k – $272k/yr
- [Applied AI Engineer - Bay Area](https://hotfix.jobs/jobs/e5aabbc0-b3cf-415e-b0ff-f96dbe33e938) - Redis - San Francisco, CA - $211k – $264k/yr
- [Research Engineer – Evals](https://hotfix.jobs/jobs/5d35303f-5667-41b7-a03c-dc6f8f99313e) - Firecrawl - Remote - $210k – $275k/yr

**Apply:** https://hotfix.jobs/jobs/eb72cd41-2cc0-499f-b1aa-f5721d12433c
**Canonical:** https://hotfix.jobs/jobs/eb72cd41-2cc0-499f-b1aa-f5721d12433c