AI Research Engineer
$214k - $285k • New York • FullTime
Posted 8mo ago
About the job
Hex is seeking an AI Research Engineer to join their mission of making everyone a data person. This role partners with product teams to build industry-leading AI experiences, such as the Notebook Agent, which assists users with complex data work by writing SQL and Python, creating reports, and exploring new avenues of inquiry. The AI Research Engineer will be instrumental in driving Hex's context engine and advancing the capabilities of the Notebook Agent, enabling data teams to build accurate and curated data experiences for their stakeholders. This is an opportunity to work on greenfield applications, shape the architecture and stack for AI-enabled capabilities, and ship product experiences that transform how Data Scientists and Analysts operate.
Responsibilities
- Run experiments, fine-tune models, and deploy AI infrastructure.
- Build and maintain experimentation tooling.
- Drive Hex's context engine and advance the capabilities of the Notebook Agent.
- Enable data teams to build highly accurate and curated data experiences.
- Build features and experiences from 0 to 1.
- Partner on determining the architecture and stack for AI-enabled capabilities.
- Ship product experiences that fundamentally change how Data Scientists and Analysts operate.
- Work at the cutting edge of production AI applications.
Requirements
- Experience getting AI/ML capabilities into production and serving real users.
- Enthusiasm for applications of AI to real business problems.
- Understanding of core MLOps/SW Architecture concepts for modern ML-based applications.
- Comfort with infrastructure aspects of MLOps.
- Comfort working in both Python and JS/TS (openness to both is important).
- Experimentalist mindset with an understanding of rapid iteration and critical thinking.
- Interest in the data space and a love of shipping great products.
- Experience maintaining a high quality bar for design, correctness, and testing.
Benefits
- Market-benched salary & equity
- Comprehensive health benefits
- Flexible paid time off