Software Engineer, Machine Learning

Key Skills

AI technology, Apache Airflow, Apache Flink, Apache Hadoop, Apache Kafka, Machine learning techniques, SaaS, SPARK Programming, YARN

Job Description

Notion is an in person company, and currently requires its employees to come to the office for two Anchor Days (Mondays & Thursdays) and requests that employees spend the majority of their week in the office (including a third day).

About The Role

We believe that recent advances in machine learning and artificial intelligence will fundamentally reshape how people use tools, and we’re looking for people who have experience in building AI-powered products to help us in forging this path. Notion’s product can be enhanced by machine learning—search ranking and relevance, assisted document composition, “smart” notifications, and abuse detection are just a few of the examples.

In this role, you’ll be responsible for end-to-end development of machine learning products—this includes building the data and machine learning infrastructure for model training and serving, training and tuning models, and working on the products that rely on those models.

What You'll Achieve:

• Depending on your interests and experience, you’ll work on fundamental infrastructure for machine learning, model training and development, ML product development, or—ideally—all of these.
• You’ll build machine learning infrastructure to enable you, your teammates, and your colleagues to train and deploy models quickly and reliably,
• You’ll develop the right models for various product use cases, and
• You’ll build the products that rely on those models, thinking carefully about the modeling and product interactions.
• You’ll work with Product, Design and engineering peers to craft our overall AI product roadmap.
• You’ll help develop our long-term technical roadmap for machine learning and artificial intelligence so we can execute on that AI product strategy.

Skills You'll Need to Bring:

• You have 7+ years of engineering experience -- ideally with the last several years focused on building machine learning-powered products and/or developing machine learning infrastructure.
• You are familiar with the fundamentals of machine learning and artificial intelligence: model types and how they’re used (everything from classical models like boosted trees to transformers and large language models). You have experience training and fine-tuning models.
• You have a track record for shipping scalable and reliable user-facing products or infrastructure.
• You thrive when collaborating cross-functionally—especially with product, design, infrastructure and data teams
• You have a computer science background or equivalent technical experience

Nice to Haves:

• You have stayed (and continue to stay) abreast of recent advances in artificial intelligence and machine learning.
• You have a Masters or Ph.D. in machine learning, artificial intelligence, or a related field.
• You have experience with large scale data processing technologies (e.g. Spark, Flink, Kafka, Airflow, YARN/Hadoop)
• You have worked at a fast-growing start-up or a SaaS company, or you’re eager to contribute in such an environment (being a current Notion user would be great!)
• You've heard of computing pioneers like Ada Lovelace, Douglas Engelbart, Alan Kay, and others—and understand why we're big fans of their work.

Our customers come from all walks of life and so do we. We hire great people from a wide variety of backgrounds, not just because it's the right thing to do, but because it makes our company stronger. If you share our values and our enthusiasm for small businesses, you will find a home at Notion.

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 made due to a disability, please let your recruiter know.

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, the estimated base salary range for this role is $220,000 - $300,000 per year.