Software Engineer, AI Platform

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

Job summary

Notion is seeking an AI Platform engineer to scale AI primitives and platforms that power Notion''s AI features. You''ll partner with product teams to ship faster while maintaining safety and reliability.

The role spans backend, infrastructure, and product code with a strong emphasis on ownership and pragmatic problem solving. The successful candidate will operate critical AI systems in production, optimize latency and cost, and contribute to versioning and controlled rollouts for new models and

Qualifications

• Experience on LLM, ML platform, data, or infrastructure teams owning shared systems.
• Willing to work across backend, infrastructure, libraries, and product code.
• Strong ownership and ability to drive execution across ambiguous spaces.
• Ability to decompose complex behavior and propose pragmatic solutions.
• Pragmatic with business focus, balancing quality and speed.

Responsibilities

• Own prototyping, development, and scaling of AI platform primitives.
• Work with product teams to provide guardrails for faster shipping.
• Collaborate across infrastructure, libraries, APIs, and product integration points.
• Operate critical AI systems in production with observability and diagnostics.
• Help Notion adopt new models and providers with clear quality gates.

Skills

AI systems

System scalability

Observability

Ownership

Cross-functional

Tools

TypeScript

Node.js

Spark

Ray

MLOps

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.