Senior Software Engineer, Machine Learning

About The Role

We are looking for a Senior Software Engineer specializing in Machine Learning to join our small but nimble AI team at Notion. Our mission is to make Notion an ML-powered product. As a Machine Learning Engineer, you will work on incorporating large language models (LLMs), embeddings, and other AI technologies into Notion’s product in a high-quality way. You’ll be exploring the boundaries of what’s possible with ML technology and finding innovative ways to apply new industry learnings to Notion’s offering.

What You'll Achieve

• Work with the team to prototype and experiment with AI model quality improvements, either by fine-tuning, prompt engineering, or building new models when needed.
• Productionize and launch new AI technology integrations into Notion’s core product.
• Collaborate with cross-functional teams to deliver product features on time.
• Stay up-to-date with the latest AI technologies and trends.

Skills You'll Need To Bring

• Domain Expert, Teacher, and Learner: You have experience building AI products using LLMs, embeddings, or other ML natural language technologies. 3+ years of experience in one or more of the following areas: machine learning, recommendation or ranking systems, natural language understanding/generation, or artificial intelligence.
• Holistic Problem Solver: You approach problems holistically, starting with a clear and accurate understanding of the context. You think critically about the implications of what you're building and how it will impact real people's lives.
• Communicate with Care: You communicate nuanced ideas clearly, whether you're explaining technical decisions in writing or brainstorming in real time. You enjoy collaborating with both fellow engineers and cross-functional partners.
• Impact Driven: You care about business impact and prioritize projects accordingly. You understand the balance between craft, speed, and the bottom line.

Nice To Haves

• Familiarity with relational database systems like Postgres or MySQL.
• Proficiency with data pipeline technologies: Spark, DBT, etc.
• Experience with our technology stack: React, TypeScript, Node.js, and Postgres.
• Experience driving teams toward shared goals and balancing business priorities with individuals’ strengths.

About Us

Notion is committed to providing highly competitive cash compensation, equity, and benefits. We hire talented and passionate people from a variety of backgrounds because we want our global employee base to represent the wide diversity of our customers. If you’re excited about a role but your past experience doesn’t align perfectly with every bullet point listed in the job description, we still encourage you to apply. If you’re a builder at heart, share our company values, and are enthusiastic about making software toolmaking ubiquitous, we want to hear from you.

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, marital status, ancestry, physical or mental disability, genetic information, veteran status, gender identity or expression, sexual orientation, or other applicable legally protected characteristic.