Software Engineer, AI Platform

About the role

The AI Platform team is responsible for building the shared foundations that let Notion ship AI products quickly and operate them safely at scale. You’ll join a team of talented engineers focused on making speed and quality compatible: reliability and availability through provider changes, quality and correctness systems like evals and release gates, observability that makes failures explainable, and shared primitives for model integrations, context management, long-running actions, and cost/performance tradeoffs.

Responsibilities

• Own and play a pivotal role in the prototyping, development and scaling of systems and core AI platform primitives.
• Partner closely with product teams to provide paved paths and production-ready guardrails that help new AI features ship faster with less duplicated work.
• Work across infrastructure, shared libraries, APIs, and product integration points to make AI platform capabilities easy to adopt and high-leverage across Notion.
• Operate critical AI systems in production, using observability and diagnostics to understand provider/model behavior, debug failures, improve latency and cost, and evolve systems with minimal user disruption.
• Help Notion safely adopt new models, providers, and AI capabilities through versioning, controlled rollouts, compatibility layers, and clear quality/reliability gates.

Requirements

Millions of people use Notion — and this number is increasing every day. That means millions of people trust us to deliver a fast, reliable, and secure experience, and we value this more than anything. We want to keep earning trust, while also continuing to amaze our users with the tools they can build in Notion.

Qualifications

• To bring passion for AI systems at scale: You’ve worked on LLM, ML platform, data, or infrastructure teams that own critical shared systems. You understand the challenges of scaling reliability, latency, cost, and quality as usage and model complexity grow.
• To care deeply about building platforms that are dependable, efficient, and easy for other engineers to use: You’re adaptable and curious, you like going deep on how systems behave in practice, especially when models, providers, and product requirements are changing quickly. You’re eager to use AI tools to work smarter and are willing to move across backend, infrastructure, libraries, and product code when that’s what the problem requires.
• To be extreme in ownership: You’re comfortable working across ambiguous problem spaces, aligning stakeholders around a clear path forward, and driving execution with accountability. You take ownership of platform outcomes including reliability, quality, adoption, and operational follow-through beyond team boundaries.
• To be thoughtful in problem-solving: For you, problem-solving starts with a clear and accurate understanding of the context. You can decompose ambiguous system behavior, debug across layers, and work toward clean, pragmatic solutions by yourself or with teammates. You’re comfortable asking for help when you get stuck.
• To be pragmatic and business-oriented: You understand that AI platform work is full of tradeoffs across quality, latency, cost, reliability, and speed of execution. You prioritize based on product and business impact, balancing craft with urgency and operational simplicity.

Skills

• Passion for AI systems at scale: Experience with applied AI product development (prompting, evals, model integrations, or quality measurement).
• Adaptable and curious: Past experience working full-stack in TypeScript and node.js ecosystem, experience building MLOps and ML serving infrastructure.

Benefits

We offer highly competitive cash compensation, equity, and benefits. The compensation offered for this role will be based on multiple factors such as location, the role's scope and complexity, and the candidate's experience and expertise, and may vary from the range provided below. For roles based in San Francisco or New York City, the estimated base salary range for this role is $180,000 - $201,000 per year.

Pay

The estimated base salary range for this role is $180,000 - $201,000 per year.

Schedule

We work from our offices on Mondays, Tuesdays and Thursdays (our Anchor Days) because we do our best thinking and building together in person.