Software Engineer, Machine Learning

1 Month ago • 3 Years + • $170,000 PA - $250,000 PA

Job Summary

Job Description

As an ML Engineer, you will integrate AI technologies like LLMs and embeddings into Notion's product. You'll experiment with AI model improvements, productionize new AI integrations, collaborate with teams, and stay updated with AI trends. The role involves prototyping and launching new AI technology, working with cross-functional teams, and staying updated with the latest AI technologies. The engineer will be responsible for ensuring new AI features are well integrated into the product and contribute to making Notion an ML-powered product, with the goal of providing a better user experience using new AI technologies.
Must have:
  • Experience building AI products using LLMs, embeddings, or other ML technologies.
  • Experience in machine learning, recommendation systems, or natural language understanding.
  • Ability to approach problems holistically and decompose complex problems.
  • Excellent communication skills both written and oral.
  • Prioritize projects accordingly based on business impact
Good to have:
  • Understanding of system components from user interface to data model.
  • Proficiency with data pipeline technologies like Spark and DBT.
  • Proficiency with the technology stack: React, TypeScript, Node.js, and Postgres.
  • Experience driving teams toward shared goals and balancing business priorities.

Job Details

About Us:

We're on a mission to make it possible for every person, team, and company to be able to tailor their software to solve any problem and take on any challenge. Computers may be our most powerful tools, but most of us can't build or modify the software we use on them every day. At Notion, we want to change this with focus, design, and craft.

We've been working on this together since 2016, and have customers like OpenAI, Toyota, Figma, Ramp, and thousands more on this journey with us. Today, we're growing fast and excited for new teammates to join us who are the best at what they do. We're passionate about building a company as diverse and creative as the millions of people Notion reaches worldwide.

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 are looking for an ML Engineer to join our small but nimble AI team whose mission is to make Notion an ML-powered product. As an ML 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. You’re excited about bringing your pre-existing knowledge to the team, but you balance that with perpetual curiosity and understanding that there is always more to learn.
  • 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. You can navigate ambiguity flawlessly, decompose complex problems into clean solutions, while also balancing the business impact of what you’re building.
  • Communicate with Care: You communicate nuanced ideas clearly, whether you're explaining technical decisions in writing or brainstorming in real time. In disagreements, you engage thoughtfully with other perspectives and compromise when needed. You enjoy collaborating with both fellow engineers and cross-functional partners. You are a lifelong learner and invest in both your own growth and the growth, learning, and development of your teammates.
  • Impact Driven: You care about business impact and prioritize projects accordingly. You understand the balance between craft, speed, and the bottom line. You think critically about the implications of what you're building, and how it shapes real people's lives. You understand that reach comes with responsibility for our impact—good and bad. Work isn't a solo endeavor for you, and you enjoy collaborating cross-functionally to accomplish shared goals.

Nice to Haves:

  • You understand how parts of a system fit together, from the user interface to the data model. You are familiar with relational database systems like Postgres or MySQL, and have experience building products from ground up.
  • You’re proficient with data pipeline technologies: Spark, DBT, etc
  • You're proficient with any part of our technology stack: React, TypeScript, Node.js, and Postgres.
  • You have experience driving teams toward shared goals and can balance business priorities with individuals’ strengths, areas of interest, and career development goals.
  • 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.
  • You have interests outside of technology, such as in art, history, or literature.

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 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 (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 and New York, the estimated base salary range for this role is $170,000 - $250,000 per year.

#LI-Onsite

By clicking “Submit Application”, I understand and agree that Notion and its affiliates and subsidiaries will collect and process my information in accordance with Notion’s Global Recruiting Privacy Policy.

 

Similar Jobs

Activision - Producer, Call of Duty

Activision

Santa Monica, California, United States (On-Site)
3 Weeks ago
Qualcomm - CPU Physical Design Engineer

Qualcomm

Santa Clara, California, United States (On-Site)
2 Weeks ago
Kaedim - DevOps Engineer

Kaedim

San Francisco, California, United States (On-Site)
9 Months ago
Agara labs - Search Engine Optimisation Expert

Agara labs

Bengaluru, Karnataka, India (Remote)
1 Month ago
Scale AI - VP of Engineering, Enterprise

Scale AI

San Francisco, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Boomi  - Senior Accountant, Financial Reporting

Boomi

Canada (On-Site)
3 Weeks ago
Techland - Creative Marketing Director

Techland

Warsaw, Masovian Voivodeship, Poland (On-Site)
4 Months ago
Riot Games - Principal Researcher - Central User Research

Riot Games

Los Angeles, California, United States (On-Site)
2 Months ago
Zelis  - Director, Product Engineering

Zelis

St. Petersburg, Florida, United States (Hybrid)
1 Month ago
hogarth - Lead QA Engineer

hogarth

Sunnyvale, California, United States (Hybrid)
1 Week ago
Tellius - Senior Software Engineer - Backend

Tellius

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Synechron - Java Spring Microservices lead

Synechron

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Philips - Program & Project support intern

Philips

Best, North Brabant, Netherlands (Hybrid)
5 Days ago
Google - Senior Software Engineer, Full Stack

Google

(On-Site)
6 Months ago
Guardian - Head Of Cyber Security Governance

Guardian

New York, United States (Hybrid)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in San Francisco, California, United States

Heart Machine - Lead / Senior Designer

Heart Machine

Los Angeles, California, United States (Remote)
2 Weeks ago
attentive - Senior Data Scientist/Applied Scientist, AI Products

attentive

San Francisco, California, United States (Hybrid)
1 Month ago
UPF Industries  - Maintenance Mechanic - 2nd Shift

UPF Industries

Thornton, California, United States (On-Site)
4 Weeks ago
Flow - Software Engineer- Backend

Flow

Palo Alto, California, United States (On-Site)
2 Months ago
Apple - Senior Business Development Manager, Apple Pay Access

Apple

New York, United States (On-Site)
3 Weeks ago
Apple - US-Manager

Apple

Littleton, Colorado, United States (On-Site)
3 Weeks ago
CD PROJEKT RED - Expert 3C's Engineer

CD PROJEKT RED

Boston, Massachusetts, United States (Remote)
2 Weeks ago
Nintendo - Contract UI Artist II

Nintendo

Redmond, Washington, United States (Hybrid)
2 Months ago
CAE - A&P Mechanics, Fixed Wing

CAE

Dothan, Alabama, United States (On-Site)
1 Week ago
Illumination - Director, Partnerships Marketing

Illumination

Santa Monica, California, United States (On-Site)
7 Months ago

Get notifed when new similar jobs are uploaded

Similar Category Jobs

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

About The Company

San Francisco, California, United States (On-Site)

New York, United States (On-Site)

San Francisco, California, United States (On-Site)

San Francisco, California, United States (On-Site)

San Francisco, California, United States (On-Site)

San Francisco, California, United States (On-Site)

San Francisco, California, United States (On-Site)

San Francisco, California, United States (On-Site)

San Francisco, California, United States (On-Site)

San Francisco, California, United States (On-Site)

View All Jobs

Get notified when new jobs are added by Notion

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug