Top AI & Machine Learning Jobs in San Francisco, CA

Analyze product changes, conduct statistical inference and experimentation, identify growth opportunities, define metrics, build datasets and dashboards, and communicate insights to influence product, engineering, marketing, and business strategy.

Conduct data analyses to identify product and business improvement opportunities, communicate actionable insights to cross-functional teams and leadership, and create metrics and dashboards monitoring Notion’s growth and health. The intern will collaborate across product and business functions, apply SQL and scripting languages such as Python or R, and prototype AI-enabled features using technologies including LLMs, embeddings, and machine learning.

The Model Behavior Engineer at Notion will ensure high-quality AI experiences through context engineering, data analysis, evaluative strategies, and collaboration with AI research labs. Responsibilities include designing tests, debugging, measuring quality, and developing systems to operationalize AI performance. This role blends product thinking with analytical skills to drive enhancements for AI tools.

As an AI Product Engineer, you will build AI-powered features for Notion's Custom Agents, enhancing workflows through LLMs and integrations.

Provide practical legal counsel for AI product development, accelerate AI vendor partnerships and model reviews, draft an AI contracting playbook, support cross-functional teams on product launches and customer AI questions, and establish a scalable AI legal operating model while adopting AI tools to improve legal workflows.

Build and scale core AI platform primitives and integrations to help product teams ship features faster and safely. Own production systems, observability, diagnostics, and reliability; manage model/provider rollouts, versioning, and quality gates. Partner across infrastructure, libraries, APIs, and product teams to improve latency, cost, and adoption.

Lead UX research to define and scale evaluation frameworks for Notion's AI experiences. Run longitudinal and feature-specific studies, create reusable rubrics and measurement approaches, identify failure modes and recovery behavior, and operationalize human-in-the-loop evaluation with product, design, engineering, and data science partners.

As an AI Applications Engineer, you'll partner with internal teams to prototype and deliver AI solutions that drive business impact, focusing on end-to-end solution development and cross-functional collaboration.

Work on AI-powered product features using LLMs and embeddings. Prototype, ship, and productionize model integrations, improve model performance/latency/cost, build evaluation frameworks, and contribute to monitoring, reliability, and UX alongside cross-functional teams.