Top AI & Machine Learning Jobs in San Francisco, CA

AI Engineer Interns will research, design, evaluate, and productionize AI-powered data capabilities. Responsibilities include running AI/ML experiments, developing agent self-improvement and continuous-learning systems, collaborating with product engineers, building evaluation and training infrastructure, optimizing performance, designing APIs, and contributing across the full development lifecycle.

Lead Hex’s AI engineering and research organization in defining strategy across post-training, model evaluation, search and retrieval, and continuous learning. Manage research partnerships, publish technical content, and hire and develop AI engineers and researchers. Partner with executive leadership and product and engineering teams to deliver AI-powered product experiences and business outcomes.

Build AI-powered features and scalable agent infrastructure for Hex’s data analytics platform. Partner with research and product engineering teams on context, model prompting, experimentation tooling, and platform integration. Lead major product initiatives, develop performant web experiences, maintain high standards for testing and correctness, and mentor engineers. The role involves full-stack development across React, TypeScript, GraphQL, backend services, cloud infrastructure, and data systems.

Build and deploy AI capabilities for Hex’s Notebook Agent and context engine. Responsibilities include running experiments, fine-tuning models, developing AI infrastructure and experimentation tooling, architecting AI-enabled systems, and shipping production experiences for data scientists and analysts. The role partners closely with product teams, spans greenfield initiatives, and offers opportunities for technical and team leadership.