AI Research Engineer

Hex•New York, NY

3d•Hybrid

About The Position

Hex is hiring a senior AI Research Engineer to partner with product teams and build AI experiences for its data science and analytics platform, including its Notebook Agent and context engine. The Notebook Agent writes SQL and Python, creates reports and visualizations, and collaborates with analysts on further analysis. The role focuses on advancing these capabilities and helping data teams create accurate, curated data experiences for their stakeholders. As an early team member, you will work across greenfield initiatives, build features from 0 to 1, help determine architecture and technology choices, and ship production AI experiences for data scientists and analysts. The position includes opportunities for technical and team leadership, depending on your interests. Hex’s web-based notebook and app authoring platform uses TypeScript and React on the frontend, with Apollo GraphQL and Redux. The backend uses TypeScript, Express, Apollo GraphQL, Postgres, Redis, and Kubernetes, with Terraform, Helm, and AWS for infrastructure and deployment. This is a full-time role based in the San Francisco or New York office, with employees encouraged to work in person at least twice a week, primarily Tuesdays and Thursdays. The salary range is $214,000–$285,000. Hex also offers equity, comprehensive health benefits, and flexible paid time off. Final salary placement depends on factors including location, skills, experience, growth opportunity, and internal fairness and parity.

Requirements

Experience getting AI/ML capabilities into production and serving real users; the role is not aimed at new graduates.
Understanding of core MLOps and software architecture concepts for modern ML-based applications.
Comfort working with Python and JavaScript/TypeScript, with openness to both even if strongest in only one.
An experimentalist mindset, with an ability to iterate quickly and apply discerning critical thinking.
Interest in the data space and enthusiasm for shipping products and building tools that empower end users.
Experience maintaining high standards for design, correctness, and testing.
Enthusiasm for applying AI to real business problems.

Nice To Haves

Strong infrastructure experience in MLOps.
Background in AI engineering, software engineering, or machine learning engineering.

Responsibilities

Run experiments and fine-tune models.
Deploy AI infrastructure and build and maintain experimentation tooling.
Drive development of Hex’s context engine and advance the capabilities of its Notebook Agent.
Build features and product experiences from 0 to 1.
Partner on determining the architecture and technology stack for AI-enabled capabilities.
Ship production AI product experiences that change how data scientists and analysts work.
Help data teams build accurate, curated data experiences for stakeholders.
Maintain a high quality bar for design, correctness, and testing.