Software Engineer, Machine Learning

About Us

Welcome to Notion! We're on a mission to empower individuals, teams, and businesses to tailor their software for any challenge. While computers are powerful tools, creating and modifying software remains out of reach for many. At Notion, we're dedicated to changing that through focus, design, and craftsmanship.

Since 2016, we've been on this journey with clients like Pixar, Mitsubishi, Figma, Plaid, and Match Group, among thousands of others. As we rapidly expand, we're excited to welcome new talents who excel in their fields. We're passionate about building a diverse and creative company that mirrors our global audience.

Notion is an in-person company. Currently, employees are required to be in the office for two Anchor Days (Mondays & Thursdays) and spend the majority of their week onsite, including a third day.

About The Role

We are seeking an ML Engineer to join our dynamic AI team. Your mission will be to integrate large language models (LLMs), embeddings, and other AI technologies into Notion's product with high quality. You'll explore the possibilities of ML technology and apply industry innovations to enhance Notion's offerings.

What You'll Achieve

• Prototype and experiment with AI model quality improvements through fine-tuning, prompt engineering, or building new models as needed.
• Productionize and launch new AI technology integrations into Notion’s core product.
• Collaborate cross-functionally to deliver product features on time.
• Stay current with the latest AI technologies and trends.

Skills You'll Need To Bring

• Domain Expert, Teacher, and Learner: With 3+ years of experience in machine learning, recommendation systems, natural language understanding/generation, or artificial intelligence, you're excited to share your expertise while remaining curious and open to learning.
• Holistic Problem Solver: You approach problems with a clear understanding of context, think critically about impact, and craft balanced solutions to complex issues.
• Communicate with Care: You clearly convey nuanced ideas, engage thoughtfully in disagreements, and collaborate effectively with engineers and cross-functional partners. Lifelong learning is key to your philosophy.
• Impact Driven: You prioritize projects with a focus on business impact, balancing craftsmanship, speed, and practical implications. Collaboration is at the heart of your work approach.

Nice To Haves

• Understanding of system integration from user interface to data model, familiarity with relational databases like Postgres or MySQL, and experience building products from the ground up.
• Proficiency in data pipeline technologies: Spark, DBT, etc.
• Experience with our technology stack: React, TypeScript, Node.js, and Postgres.
• Proven ability to drive teams towards shared goals while balancing business priorities with individual strengths and development goals.
• Knowledge of computing pioneers like Ada Lovelace, Douglas Engelbart, Alan Kay, and an appreciation for their work.
• Interests outside technology, such as art, history, or literature.

We hire passionate and talented individuals from diverse backgrounds. If you're excited about this role but your experience doesn't align perfectly with every requirement, we encourage you to apply. If you're a problem solver who shares our values and is passionate about making software accessible to all, we'd love to hear from you.

Notion is proud to be an equal opportunity employer. We do not discriminate based on race, color, religion, national origin, age, sex (including pregnancy, childbirth, or related conditions), marital status, ancestry, disability, genetic information, veteran status, gender identity or expression, sexual orientation, or other protected characteristics. We consider qualified applicants with criminal histories according to applicable law and provide reasonable accommodations for qualified individuals with disabilities and disabled veterans. If you need assistance due to a disability, please let your recruiter know.

We offer competitive compensation, equity, and benefits. The compensation for this role will be based on factors like location, role complexity, and candidate experience. For roles based in San Francisco and New York, the estimated base salary range is $170,000 - $300,000 per year.