AI Engineer (Search and Context)

Posted 12 Days Ago
3 Locations
Remote or Hybrid
225K-275K Annually
Senior 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
AI Engineers at Hex will build AI experiences, manage search and context architecture, and work on scalable AI applications. Responsibilities include running experiments, developing infrastructure, and leading initiatives for new capabilities.
Summary Generated by Built In
About the role

Hex is an AI-powered platform for modern Data Science and Data Analytics workflows. AI 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 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.

Note: We deeply respect and appreciate deep AI/ML research; this role is not that. While experience producing relevant journal publications is awesome, it won’t be part of this role’s deliverables.

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.


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.

Top Skills

AI
JavaScript
Ml
Mlops
Python
Typescript
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

Infra Engineer, Observability Whisperer

Artificial Intelligence • Big Data • Software • Analytics • Business Intelligence • Big Data Analytics
Remote or Hybrid
3 Locations
160 Employees

Hex Logo Hex

Mid-market Account Executive

Artificial Intelligence • Big Data • Software • Analytics • Business Intelligence • Big Data Analytics
Remote or Hybrid
3 Locations
160 Employees
220K-240K Annually

Hex Logo Hex

Software Engineer

Artificial Intelligence • Big Data • Software • Analytics • Business Intelligence • Big Data Analytics
Remote or Hybrid
3 Locations
160 Employees
176K-220K Annually

Hex Logo Hex

Infrastructure Engineer

Artificial Intelligence • Big Data • Software • Analytics • Business Intelligence • Big Data Analytics
Remote or Hybrid
3 Locations
160 Employees
215K-270K Annually

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account