inhousefyi
← Back to listings

AI Research Engineer

Hex TechnologiesSan Francisco, California, United States · Posted 6 months ago
Full-timeEst. 249,500 USD
Apply now

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 Research Engineers will run experiments, fine-tune models, deploy AI infrastructure, and build and maintain experimentation tooling.

As an AI Research Engineer partnered with Hex's product teams, your focus will be driving Hex's context engine and pushing forward the capabilities of our Notebook Agent. Built for people doing the most complex and impactful data work in an organization, the Notebook Agent is a highly technical data copilot that writes SQL and python, builds beautiful reports and visualizations, and collaborates with the analyst on new avenues of inquiry. Whether its pushing the frontiers of SQL generation quality or using LLMs to evaluate agent threads, your job will be to enable the data teams that work in Hex to build out highly accurate and curated data experiences for their stakeholders, unlocking the power of data driven decisions for the rest of the business.

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:

  • Building features and experiences from 0 to 1
  • Partnering on determining the architecture and stack for our AI-enabled capabilities
  • Shipping product experiences that fundamentally change the way that Data Scientists and Analysts operate
  • Working at the cutting edge of production AI applications

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 getting AI/ML capabilities into production and serving real users – we’re not currently looking for new grad folks.
  • A lot of enthusiasm for applications of AI to real business problems.
  • Understanding of core MLOps/SW Architecture concepts for modern ML-based applications.
    • Ideal candidates are strong on Infra aspects of MLOps as well.
  • 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.


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.

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: $214,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.

Hex Technologies uses AI-assisted tools as part of our application review process, including for resume screening and fraud detection. These tools help our team evaluate applications and verify applicant information. All AI-generated recommendations are reviewed by a member of our recruiting team before any hiring decision is made. No application is automatically rejected based solely on an AI tool's output.

Similar jobs

Hex TechnologiesSan Francisco, California, United States

Est. 287,500 USD

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…

Full-time
Hex TechnologiesSan Francisco, California, United States

Est. 246,875 USD

About the role We are looking for senior and staff-level fullstack engineers eager to join the team building our leading Magic AI features. At Hex we're deeply focused on building powerful, integrated AI features that au…

Full-time
Hex TechnologiesNew York City, New York, United States

Est. 246,875 USD

About the role At Hex we are using AI to reinvent how business users interact with data, enabling rich data exploration using natural language. Our team of fullstack engineers are responsible for the critical workflows t…

Full-time

Est. 246,875 USD

About the role We are looking for experienced, backend-leaning product engineers to help expand the capabilities of our industry-leading AI analytics platform. At Hex, we're deeply focused on building powerful, integrate…

Full-timeRemote
Hex TechnologiesNew York, New York, United States

Est. 305,250 USD

About the role We are looking for an experienced engineering leader to drive the strategy, execution, and growth of Hex’s state-of-the-art AI context platform. Our ambition is simple but powerful: Hex should get better e…

Full-time
Hex TechnologiesNew York City, New York, United States

Est. 135,000 USD

. About the role: Hex is scaling quickly, and the quality of our team is the single biggest driver of that. We're looking for a Technical Sourcer who treats sourcing as a craft and a system, someone who can build repeata…

Full-time
Engineering Lead1 month ago
Hex TechnologiesSan Francisco, California, United States

Est. 258,000 USD

The role We are looking for a hands-on engineering manager to drive the strategy, execution, and growth of Hex’s state-of-the-art AI context platform. Our ambition is simple but powerful: Hex should get better every time…

Full-time
Product ExpertAug 13, 2025

Est. 120,000 USD

About the Role Hex is growing fast; we are hiring a Product Expert to nurture and support customers to make them successful with Hex. Our focus is on building a valuable product that users love, and all your work will fe…

Full-timeRemote

Est. 284,875 USD

We are looking for a Staff or Principal-level software engineer to join our Compute team. Hex's Compute team owns one of the hardest performance problems in the product: our users do everything from analyzing 100-line CS…

Full-timeRemote

Est. 231,875 USD

The role We’re looking for an experienced full-stack Software Engineer to join our Growth & Monetization group. This group spans several work streams — including product-led growth (PLG), user activation, pricing and…

Full-timeRemote

Est. 207,375 USD

About the team The Foundations team builds and evolves the application-level infrastructure that powers Hex. We own the shared backend systems, frameworks, and primitives that other teams depend on—helping the company mo…

Full-timeRemote
Hex TechnologiesNew York City, New York, United States

Est. 135,200 USD

About the role: As Recruiting Operations Manager, you'll own the systems, processes, data, and programs that power hiring at Hex. This is a builder role. You'll take a recruiting function that's growing fast and give it…

Full-time
Customer Engineer11 months ago
Hex TechnologiesSan Francisco, California, United States

Est. 185,500 USD

=== Page: https://hexhq.notion.site/the-customer-team-hex?pvs=143 === About the Role Hex's customer base is growing and changing quickly as our space and product evolve. To keep pace with this growth, we are hiring the n…

Full-time

Est. 237,500 USD

Hex's user base is continuing to grow rapidly. We are hiring a Sales Engineer to partner with potential customers to support their Hex evaluations. About the role Hex is redefining how teams work with data - combining no…

Full-timeRemote
Hex TechnologiesSan Francisco, California, United States

Est. 238,500 USD

About the role Hex is looking for a Product Manager to join the team and help drive the future of how teams work with data. Hex is a product-led company with a strong sales motion, so you’ll get to think deeply about pro…

Full-time
Hex TechnologiesNew York City, New York, United States

Est. 238,500 USD

About the role Hex is looking for a Product Manager to join the team and help drive the future of how teams work with data. Hex is a product-led company with a strong sales motion, so you’ll get to think deeply about pro…

Full-time
Hex TechnologiesSan Francisco, California, United States

Est. 180,000 USD

About the Role In August of 2021, Hex began building out its Sales team. Since then our customer base has grown, we have a ton of momentum in the market, and we’re continuing to build a world-class sales organization. We…

Full-time

Est. 232,500 USD

About the role We are looking for an experienced Cloud Security Engineer to join Hex’s security team. You will be responsible for ensuring the security and resilience of our cloud infrastructure, providing leadership in…

Full-timeRemote
Hex TechnologiesNew York City, New York, United States

Est. 212,500 USD

About the role Hex is looking for a Senior Product Manager, Growth to join our Growth & Monetization group. This group spans several work streams, including product-led growth (PLG), new user activation, pricing and…

Full-time

Est. 320,000 USD

About the Role In August of 2021, Hex began building out its Sales team. Since then our customer base has grown, we have a ton of momentum in the market, and we’re continuing to build a world-class sales organization. We…

Full-timeRemote
Web Designer8 months ago

Est. 205,000 USD

=== Process: https://hexhq.notion.site/Hex-s-Web-Designer-Interview-Process-2ee45d6bfe7c8057ba69fcf715b02812 TeamPage: https://hexhq.notion.site/Design-Hex-21a4d4a08e23457faad1378455e18ac0 === Members of the Brand Design…

Full-timeRemote

Est. 250,000 USD

About the Role In August of 2021, Hex began building out its Sales team. Since then our customer base has grown, we have a ton of momentum in the market, and we’re continuing to build a world-class sales organization. We…

Full-timeRemote

Est. 206,750 USD

About the role: This is a ground floor opportunity to build the next generation of our partner function at Hex. Our partnerships team is a mission critical aspect of our GTM strategy. This role represents all things tech…

Full-timeRemote

Est. 250,000 USD

About the Role In August of 2021, Hex began building out its Sales team. Since then our customer base has grown, we have a ton of momentum in the market, and we’re continuing to build a world-class sales organization. We…

Full-timeRemote
SDR Manager1 month ago
Hex TechnologiesSan Francisco, California, United States

Est. 177,500 USD

About the Role We're hiring an SDR Manager to lead and scale our West Coast SDR team, at a real inflection point for the function, the team is shifting from an inbound-reliant motion to one that generates its own outboun…

Full-time
Hex TechnologiesSan Francisco, California, United States

Est. 100,000 USD

About the Role In August of 2021, Hex began building out its Sales team. Since then our customer base has grown, we have a ton of momentum in the market, and we’re continuing to build a world-class sales organization. We…

Full-time
AlphaSenseSan Francisco, California, United States

Est. 253,500 USD

About AlphaSense: The world’s most sophisticated companies rely on AlphaSense to remove uncertainty from decision-making. With market intelligence and search built on proven AI, AlphaSense delivers insights that matter f…

Full-time
Sigma ComputingSan Francisco, California, United States

Est. 255,000 USD

About the Role At Sigma, we’re not just adding AI—we’re building the future of how people work with data. Our platform already lets users explore billions of rows of data in seconds with a spreadsheet-like interface, ana…

Full-time
AlphaSenseNew York, New York, United States

Est. 253,500 USD

About AlphaSense: The world’s most sophisticated companies rely on AlphaSense to remove uncertainty from decision-making. With market intelligence and search built on proven AI, AlphaSense delivers insights that matter f…

Full-time
Sigma ComputingNew York City, New York, United States

Est. 255,000 USD

About the Role At Sigma, we’re not just adding AI—we’re building the future of how people work with data. Our platform already lets users explore billions of rows of data in seconds with a spreadsheet-like interface, ana…

Full-time