AI Engineer – Search, Context

Job not on LinkedIn

🕒 December 29, 2025

🏢🏡 New York City – Hybrid

💵 $225k - $275k / year

⏰ Full Time

🟡 Mid-level

🟠 Senior

🤖 AI Engineer

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

Apply Now

Find Similar Remote Jobs

Receive Emails For Remote Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

Hex

WebsiteLinkedInAll Job Openings

51 - 200 employees

🤖 Artificial Intelligence

☁️ SaaS

Artificial Intelligence • Data Analytics • SaaS

Hex is a collaborative, AI-powered workspace that brings teams together with data. It enables end-to-end workflows, from quick queries to deep-dive analyses, and facilitates the creation of interactive data apps. Hex integrates various tools such as SQL, Python, R, and spreadsheets in a modular, notebook-based canvas, allowing users to generate queries, create visualizations, and perform analyses with ease. The platform supports teamwork by allowing feedback from peers, alignment with stakeholders, and the building of reusable components. Hex also offers a drag-and-drop builder for creating interactive reports and dashboards. With built-in connections to popular data warehouses and databases, Hex simplifies data integration and collaboration for thousands of teams. Its deep integration with tools like dbt, Snowpark for Snowflake, and orchestration frameworks like Airflow enhances its functionality in data science and business intelligence. Hex is designed for security and offers enterprise-grade controls, making it a trusted platform for data teams worldwide.

📋 Description

• Partner with product teams to build industry-leading AI experiences • Run experiments, fine-tune models, deploy AI infrastructure, and maintain experimentation tooling • Experiment with new agentic techniques for search, discovery, and context management • Design and implement the architecture for scalable search and indexing pipelines • Work at the cutting edge of production AI applications deployed to real customers

🎯 Requirements

• 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.

🏖️ Benefits

• Health benefits • Flexible paid time off • Market-benched salary & equity