Skip to content
HexHexSan Francisco, CA

AI Research Engineer

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.

214k – 285k
Hybrid5+ YOEML Engineering

About the role

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.

Skills

PythonTypeScriptReactKubernetesMLOpsGraphQLPostgresRedisTerraformAWS

Similar roles

ML Engineering jobs
Liftoff

Machine Learning Engineer

LiftoffCalifornia

Machine Learning Engineer building statistical models, optimization systems, and experiments for mobile ad tech economics on the Revenue Engine team. Requires PhD in CS/ML/Economics and industry experience applying ML or economics at scale.

215k – 275k
RemoteML Engineering
Scale AI

Software Engineer, Enterprise AI

Scale AINew York, NY +1

Build and scale enterprise Generative AI platform, owning large product areas across backend, frontend, LLMs, and ML models. Requires 4+ years experience, proficiency in Python/JavaScript/SQL, Kubernetes, and cloud providers.

216k – 270k
On-site4+ YOEML Engineering
Plaid

Machine Learning Engineer - Embedded Insights

PlaidSan Francisco, CA +2

Drive ML initiatives from concept to production on the Embedded Insights team. Identify opportunities, build and deploy models using Plaid's financial datasets, and partner with product teams to deliver scalable customer-facing intelligence products.

212k – 272k
Hybrid5+ YOEML Engineering
Redis

Applied AI Engineer - Bay Area

RedisSan Francisco, CA

Senior Applied AI Engineer focused on helping customers build AI apps with Redis (80% customer engineering, 20% development). Requires 5+ years Python experience and 2+ years in ML systems.

211k – 264k
On-site5+ YOEML Engineering
Firecrawl

Research Engineer – Evals

FirecrawlSan Francisco, CA

Build and own evaluation systems, metrics, pipelines, and datasets to rigorously measure the quality of Firecrawl's LLM-ready web data outputs at massive scale, driving model and product improvements.

210k – 275k
Remote4+ YOEML Engineering