Software Engineer, AI Platform

Posted Yesterday
2 Locations
Hybrid
180K-201K Annually
Mid level
Artificial Intelligence • Productivity • Software
Notion is the AI workspace where teams and AI think together. Come build work that lasts.
The Role
Build and scale core AI platform primitives and integrations to enable rapid, safe AI product development. Own infrastructure, shared libraries, APIs, observability, and reliability systems; operate production AI systems, debug provider/model behavior, and implement versioning, rollout controls, and quality gates to minimize user disruption.
Summary Generated by Built In
Who We Are

Notion is the collaborative AI workspace where teams and agents think together. We're building 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.

Notinos (our employees) are customer zero in bringing this future of work to life. We care about craft, building things that last, and the belief that great work is still fundamentally human. Our goal isn’t to ship the next feature. Each and every team of Notinos is working to set the standard for how humans work together in the AI era. From building a business’s system of record to making and managing AI agents to automating away the busy work, we care deeply about giving our customers more time for their life’s work.

About the Role:

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.

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. Notion’s AI platform is vital to helping product teams move faster with production-grade guardrails as models, providers, and AI capabilities rapidly evolve.

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

What You'll Achieve:
  • You'll own and play a pivotal role in the prototyping, development and scaling of systems and core AI platform primitives.

  • You’ll partner closely with product teams to provide paved paths and production-ready guardrails that help new AI features ship faster with less duplicated work.

  • You’ll work across infrastructure, shared libraries, APIs, and product integration points to make AI platform capabilities easy to adopt and high-leverage across Notion.

  • You’ll 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.

  • You’ll 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: 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. You care deeply about building platforms that are dependable, efficient, and easy for other engineers to use.

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

  • Extreme 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.

  • Thoughtful 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.

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

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

  • You've built out and scaled data processing pipelines at scale with Apache Spark or Ray.

  • You’ve past experience working full-stack in Typscript and node.js ecosystem

  • You have experience building MLOps and ML serving infrastructure.

Notion is committed to providing 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.

By clicking “Submit Application”, I understand and agree that Notion and its affiliates and subsidiaries will collect and process my information in accordance with Notion’s Global Recruiting Privacy Policy and NYLL 144.

#LI-Onsite

A Note on AI

You don’t need deep AI expertise for every role, but we do expect every Notino to be intellectually curious, drawn to tinkering and discovery, and excited to use AI as a real collaborator in their work. For some roles, AI fluency is a core requirement — when that’s the case, we'll say so explicitly in the qualifications. People who thrive here don’t treat AI as a novelty. They use it to think better, and make their work easier for others to build on.

Equal Opportunity & Accommodations

We hire talented people from a wide range of backgrounds. If you’re excited about this role but don’t meet every bullet, we still encourage you to apply. Notion is an equal opportunity employer and does not discriminate on the basis of any legally protected characteristic. Consistent with applicable law, we will consider for employment qualified applicants with arrest and conviction records. Notion provides reasonable accommodations during the application process; if you need one, please let your recruiter know.

Notion is proud to be an equal opportunity employer. We do not discriminate in hiring or any employment decision based on race, color, religion, national origin, age, sex (including pregnancy, childbirth, or related medical conditions), marital status, ancestry, physical or mental disability, genetic information, veteran status, gender identity or expression, sexual orientation, or other applicable legally protected characteristic. Notion considers qualified applicants with criminal histories, consistent with applicable federal, state and local law. 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.

Skills Required

  • Experience owning or building AI/LLM platform, ML platform, or shared infrastructure
  • Operate production AI systems with observability, diagnostics, and debugging
  • Design and build APIs, libraries, and primitives for product teams to adopt
  • Drive reliability, availability, cost, and performance tradeoffs with production-grade guardrails
  • Strong problem-solving, ability to work across backend, infrastructure, libraries, and product code
  • Comfort working onsite in San Francisco or New York City on Anchor Days (Mon, Tue, Thu)
  • 2-4 years of experience as a Software Engineer
  • Experience with applied AI product development (prompting, evals, model integrations, quality measurement)
  • Experience building data processing pipelines with Apache Spark or Ray
  • Experience with full-stack TypeScript and Node.js ecosystem
  • Experience building MLOps and ML serving infrastructure

What the Team is Saying

Alma
Penny
Marlene

Notion Compensation & Benefits Highlights

  • Healthcare Strength Medical, dental, and vision coverage have premiums covered in most cases, and mental‑health support is included.
  • Parental & Family Support Paid parental leave (including adoptive and foster paths) is provided, alongside employer‑funded fertility and family‑forming support via Carrot.
  • Equity Value & Accessibility An employee tender and a temporary waiver of the one‑year vesting cliff enabled broad participation, improving perceived liquidity and access to equity value.

Notion Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, CA
1,000 Employees
Year Founded: 2016

What We Do

Notion is a collaborative AI workspace. Teams use it to store knowledge, run projects, take meeting notes, and build AI workflows — all in one place, so AI always has the context it needs to be useful. We're building the layer where AI stops being a chat window and starts being part of how work gets done. Knowledge, decisions, and action in one connected system.

Why Work With Us

We're building the layer where AI stops being a chat window and starts being part of how work gets done. Notion is where AI becomes useful at work: real workflows, real users, real context. People here move from idea to shipped without asking permission. If you want your fingerprints on work that millions of people use, this is the place.

Gallery

Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery

Notion Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

Notion is an in-person company. We come together three days a week: Monday, Tuesday, and Thursday — because the best collaboration tends to sneak up on you. A quick question becomes a whiteboard session. A lunch reshapes a plan. Togetherness is what

Typical time on-site: 3 days a week
Company Office Image
HQSan Francisco, CA
Company Office Image
Dublin, Dublin
Company Office Image
Hanyang, KR
Company Office Image
Hyderabad, Hyderabad
Company Office Image
New York, NY
Company Office Image
Tokyo, Tokyo
Learn more

Similar Jobs

Notion Logo Notion

Solutions Engineer

Artificial Intelligence • Productivity • Software
Hybrid
2 Locations
1000 Employees
220K-260K Annually

Notion Logo Notion

Solutions Engineer

Artificial Intelligence • Productivity • Software
Hybrid
2 Locations
1000 Employees
220K-260K Annually

Notion Logo Notion

Technical Enablement Manager, Customer Experience

Artificial Intelligence • Productivity • Software
Hybrid
New York, NY, USA
1000 Employees
165K-180K Annually

Notion Logo Notion

Architect

Artificial Intelligence • Productivity • Software
Hybrid
New York, NY, USA
1000 Employees
165K-180K Annually

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account