AI Research Engineer, Search and Context

at Hex · Data AI · New York, NY +1 · Engineering

AI Research Engineer focused on search and context for an AI-powered data science platform. Responsibilities include building agentic platform components, agentic search/discovery subagents, and high-scale indexing systems. The role involves experimenting with agentic techniques, designing scalable pipelines, and deploying AI to customers.

What you'd actually do

  1. Experimenting with new agentic techniques for search, discovery, and context management
  2. Designing and implementing the architecture for our scalable search and indexing pipelines
  3. Working at the cutting edge of production AI applications deployed to real customers

Skills

Required

  • Python
  • JS/TS
  • building and measuring high quality search and recommendation systems
  • getting AI/ML capabilities into production and serving real users
  • MLOps/SW Architecture concepts for modern ML-based applications
  • Experimentalist mindset
  • shipping great products
  • building tools that empower end users
  • maintaining a high quality bar for design, correctness, and testing

Nice to have

  • openness to both Python & JS/TS

What the JD emphasized

  • building out key components of our agentic platform
  • agentic search and discovery subagents
  • high-scale, permissions-aware indexing systems
  • production AI applications deployed to real customers
  • senior engineer
  • early hires
  • greenfield applications
  • ownership
  • technical and team leadership opportunities
  • radically accelerate and augment the data science workflow
  • inspired excitement about what is possible in this space
  • Experience building and measuring high quality search and recommendation systems
  • Experience getting AI/ML capabilities into production and serving real users
  • Understanding of core MLOps/SW Architecture concepts for modern ML-based applications.
  • Experimentalist mindset
  • shipping great products and building tools that empower end users to do more.
  • maintaining a high quality bar for design, correctness, and testing.

Other signals

  • building out key components of our agentic platform
  • agentic search and discovery subagents
  • high-scale, permissions-aware indexing systems
  • production AI applications deployed to real customers
Read full job description

About the role

Hex is an AI-powered platform for modern Data Science and Data Analytics workflows. AI Research Engineers at Hex partner with product teams to build industry-leading AI experiences such as the Notebook Agent. In the process, AI Engineers will run experiments, fine-tune models, deploy AI infrastructure, and build and maintain experimentation tooling.

The backbone of all of our AI experiences is providing relevant context to the agent. As an AI Research Engineer focused on our search and context architecture, you’ll be responsible for building out key components of our agentic platform, from agentic search and discovery subagents to high-scale, permissions-aware indexing systems.

If you’re a strong builder excited to bring these capabilities to thousands of users, come do it on the best Data Science platform around, with the best user context.

We are looking for a senior engineer (from an AI Eng, SWE or MLE background) eager to rapidly expand our capabilities in several greenfield applications. As one of our team’s early hires, you will work on a variety of initiatives including:

  • Experimenting with new agentic techniques for search, discovery, and context management
  • Designing and implementing the architecture for our scalable search and indexing pipelines
  • Working at the cutting edge of production AI applications deployed to real customers

This type of ownership will provide incredible opportunities for personal and professional growth. You will have ample technical and team leadership opportunities, depending on your interests.

We aim to deliver AI capabilities that radically accelerate and augment the data science workflow – we’ve got an exciting roadmap and are willing to share more details during the process. We’re particularly interested in candidates who bring inspired excitement about what is possible in this space.

We'd love to hear from you if you have:

  • Experience building and measuring high quality search and recommendation systems
  • Experience getting AI/ML capabilities into production and serving real users
  • A lot of enthusiasm for applications of AI to real business problems.
  • Understanding of core MLOps/SW Architecture concepts for modern ML-based applications.
  • You’re comfortable working in both Python & JS/TS – it’s okay if you’re only strong in one, but openness to both is important.
  • Experimentalist mindset – understanding that quick iteration, and discerning critical thinking, can yield rapid progress on new challenges.
  • Interest in the data space, and a love of shipping great products and building tools that empower end users to do more.
  • Experience maintaining a high quality bar for design, correctness, and testing.

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.

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: $225,000-$285,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.