Software Engineer, AI Platform

Notion; San Francisco (CA); Hybrid; USD 180,000 - 201,000; Full time

Job summary

Notion is seeking a software engineer to build core AI platform primitives and scale AI infrastructure for reliable, safe operation at scale. You\'ll partner with product teams to ship AI features faster with guardrails and production-ready components.

You\'ll work across infrastructure, shared libraries, APIs, and model integrations to make AI capabilities easy to adopt and high-leverage across Notion. This role requires ownership, pragmatic problem solving, and the ability to balance quality,

Qualifications

• Passion for AI systems at scale, experience on LLM/ML platform teams that own critical shared systems.
• Willingness to dig into system behavior as models and requirements evolve.
• Ability to align stakeholders around a path forward and execute with accountability.
• Ability to decompose complex system behavior and deliver pragmatic solutions.
• Understand tradeoffs among quality, latency, cost, and reliability in a product context.

Responsibilities

• Prototype, develop, and scale AI platform primitives.
• Collaborate with product teams to provide paved paths and guardrails for faster AI feature delivery.
• Work across infra, shared libraries, APIs, and product integration points for easy adoption.
• Operate production AI systems with observability and diagnostics to reduce latency and cost.
• Support safe adoption of new models/providers through versioning and release gates.

Skills

AI systems at scale

Adaptable and curious

Extreme ownership

Thoughtful problem solving

Pragmatic and business oriented

Tools

TypeScript

Node.js

Apache Spark

Ray

MLOps infrastructure

Job description

Notion is the collaborative AI workspace where teams and agents think together. We build one place where your knowledge, projects, meetings, and AI tools live side by side, so work is faster, clearer, and less fragmented. Millions of individuals, small teams, and large companies run their work on Notion. Our employees, or "Notinos," are customer zero, shaping the future of work and striving to give our customers more time for their life’s work.

About the Role

Millions of people use Notion, and that number is increasing every day. You’ll build the shared foundations that let us ship AI products quickly and operate them safely at scale. 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.

This role can be based in either San Franklin or New York City. We work from our offices on Mondays, Tuesdays, and Thursdays, because we do our best thinking and building together in person. We’re looking for someone excited to work alongside the team during those days.

What You’ll Achieve

• Own and play a pivotal role in prototyping, developing, and scaling 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.

Skills You’ll Need to Bring

• Passion for AI systems at scale: experience on LLM, ML platform, data, or infrastructure teams that own critical shared systems, and a deep understanding of scaling reliability, latency, cost, and quality.
• Adaptable and curious: a willingness to dig deep into how systems behave in practice, especially as models, providers, and product requirements change quickly.
• Extreme ownership: comfort working across ambiguous problem spaces, aligning stakeholders around a clear path forward, and driving execution with accountability.
• Thoughtful problem‑solving: decompose ambiguous system behavior, debug across layers, and produce clean, pragmatic solutions, either independently or with teammates.
• Pragmatic and business‑oriented: understand tradeoffs across quality, latency, cost, reliability, and speed of execution, and prioritize based on product and business impact.

Nice to Haves

• 2–4 years of experience as a software engineer.
• Experience with applied AI product development (prompting, evals, model integrations, or quality measurement).
• Previous experience building and scaling data processing pipelines at scale with Apache Spark or Ray.
• Full‑stack experience in the TypeScript and Node.js ecosystem.
• Experience building MLOps and ML serving infrastructure.

Notion is committed to providing highly competitive cash compensation, equity, and benefits. The compensation for this role is based on multiple factors such as location, scope, and candidate experience. For roles based in San Francisco or New York City, the estimated base salary range is $180,000 – $201,000 per year.

A Note on AI

Deep AI expertise is not required for every role, but we expect every Notino to be intellectually curious, eager to tinker, and excited to use AI as a real collaborator. For some roles, AI fluency is a core requirement, and we’ll state that explicitly in the qualifications. People who thrive here use AI to think better and to make their work easier for others to build on.

Equal Opportunity & Accommodations

We hire talented people from a wide range of backgrounds. Notion is an equal‑opportunity employer and does not discriminate on the basis of any legally protected characteristic. We consider qualified applicants with arrest and conviction records, and we provide reasonable accommodations during the application process. If you need an accommodation, please let your recruiter know.

Notion is also committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. If you need assistance or an accommodation due to a disability, please let your recruiter know.