Notion

Software Engineer, AI Platform

Notion
Apply
2 months ago
San Francisco, CA, USA or New York, NY, USAMid Level
H1B Sponsor

Base Salary

$180k - $201k/yr

Responsibilities

  • Prototype, develop, and scale systems and core AI platform primitives.
  • Partner with product teams to provide paved paths and production-ready guardrails for AI feature development.
  • Build across infrastructure, shared libraries, APIs, and product integration points to make AI capabilities broadly adoptable.
  • Operate critical AI systems in production and use observability and diagnostics to debug failures and improve latency and cost.
  • Support adoption of new models, providers, and AI capabilities through versioning, controlled rollouts, compatibility layers, and quality and reliability gates.

Requirements

  • Passion for AI systems at scale and experience working on LLM, ML platform, data, or infrastructure teams that own critical shared systems.
  • Ability to work across backend, infrastructure, shared libraries, and product code while debugging complex system behavior.
  • Strong ownership, problem-solving, stakeholder alignment, and execution skills in ambiguous technical environments.
  • Experience balancing quality, latency, cost, reliability, and execution speed in AI platform work.
  • Preferred: 2-4 years of experience as a Software Engineer.
  • Preferred: experience with applied AI product development, including prompting, evaluations, model integrations, or quality measurement.
  • Preferred: experience scaling data processing pipelines with Apache Spark or Ray.
  • Preferred: full-stack experience with TypeScript and the Node.js ecosystem.
  • Preferred: experience building MLOps and ML serving infrastructure.

Benefits

  • Competitive cash compensation, equity, and benefits are offered.
  • The role is based in San Francisco or New York City.
  • Employees work from the office on Mondays, Tuesdays, and Thursdays as Anchor Days.
Notion

About Notion

501-1,000 employees

Notion blends your everyday work tools into one. Product roadmap? Company wiki? Meeting notes? With Notion, they're all in one place, and totally customizable to meet the needs of any workflow. It's the all-in-one workspace for you, your team, and your whole company. We humans are toolmakers by nature, but most of us can't build or modify the software we use every day — arguably our most powerful tool. Our team at Notion is on a mission to make it possible for everyone to shape the tools that shape their lives.

Contact me