Software Engineer (AI Platform)

Join Notion, a rapidly growing platform trusted by millions of users. As a Software Engineer on the AI Platform team, you'll play a pivotal role in building the foundations for AI products, ensuring speed, quality, and reliability. You'll work closely with product teams, operate critical AI systems in production, and help Notion safely adopt new models and capabilities. Ideal candidates have experience in AI systems at scale, extreme ownership, and a pragmatic, business-oriented mindset.

• Prototyping, developing, and scaling systems and core AI platform primitives.
• Partnering closely with product teams to provide paved paths and production-ready guardrails.
• Operating critical AI systems in production, using observability and diagnostics to understand provider/model behavior.
• 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
• 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
• 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
• 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
• 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
• You've built out and scaled data processing pipelines at scale with Apache Spark or Ray
• You have experience building MLOps and ML serving infrastructure
• Experience with applied AI product development (prompting, evals, model integrations, or quality measurement)
• You’ve past experience working full-stack in Typscript and node.js ecosystem
• 2-4 years of experience as a Software Engineer