Software Engineer, AI Platform

Job description

About the Role

Millions of people use Notion and this number is increasing every day. Notion relies on delivering a fast, reliable, and secure experience. The AI Platform team builds shared foundations that allow Notion to ship AI products quickly and operate them safely at scale.

This role can be based in either San Francisco or New York City. The team works from our offices on Mondays, Tuesdays and Thursdays (Anchor Days) and expects team members to work alongside the team during those days.

What You'll Achieve

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.

Skills You'll Need to Bring

Passion for AI systems at scale: have worked on LLM, ML platform, data, or infrastructure teams that own critical shared systems and understand scaling challenges for reliability, latency, cost, and quality.

Adaptable and curious: like deep digging into system behavior when models, providers, and product requirements change quickly, and are willing to move across backend, infrastructure, libraries, and product code as needed.

Extreme ownership: comfortable working across ambiguous spaces, aligning stakeholders around a clear path forward, and driving execution with accountability.

Thoughtful problem‑solving: start with a clear and accurate understanding of context, decompose ambiguous behavior, debug across layers, and propose pragmatic solutions.

Pragmatic and business‑oriented: understand trade‑offs across quality, latency, cost, reliability, and speed, prioritize based on product and business impact, and balance craft with urgency and operational simplicity.

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).

Built out and scaled data processing pipelines at scale with Apache Spark or Ray.

Past experience working full‑stack in TypeScript and Node.js ecosystem.

Experience building MLOps and ML serving infrastructure.

Benefits

Notion is committed to providing highly competitive cash compensation, equity, and benefits. For roles based in San Francisco or New York City, the estimated base salary range is $180,000 – $201,000 per year.

Note on AI

You don’t need deep AI expertise for every role, but we expect every Notino to be intellectually curious, drawn to tinkering and discovery, and excited to use AI as a real collaborator in their work. People who thrive here don’t treat AI as a novelty; they use it to think better and to make their work easier for others to build on.

Equal Opportunity & Accommodations

Notion is an equal opportunity employer and does not discriminate on the basis of any legally protected characteristic. We consider qualified applicants with criminal histories, consistent with applicable federal, state, and local law, and provide reasonable accommodations during the application process. Notion is committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in its job application procedures.

#J-18808-Ljbffr

Worksite address

san francisco, CA, 94199, US

Who can apply

Review the original listing for work authorization, qualifications and employer requirements.