Staff ML Engineer, Applied AI

Amplitude is the leading digital analytics platform that helps companies unlock the power of their products. Over 4,000 customers, including Atlassian, NBCUniversal, Under Armour, Shopify, and Jersey Mike’s, rely on Amplitude to gain self-service visibility into the entire customer journey. Amplitude guides companies every step of the way as they capture data they can trust, uncover clear insights about customer behavior, and take faster action. When teams understand how people are using their products, they can deliver better product experiences that drive growth. Amplitude is the best-in-class analytics solution for product, data, and marketing teams, ranked #1 in multiple categories in G2’s Spring 2025 Report. Learn how to optimize your digital products and business at amplitude.com.

As an organization, we approach challenges with humility, take ownership of our contributions, and embrace a growth mindset that pushes us to constantly improve ourselves, each other, and the value we bring to customers and partners.

Amplitude’s Commitment to Diversity Equity & Inclusion (DEI): Amplitude believes that diversity enables the creation of better products, improves the ability to solve complex problems, and drives more powerful solutions. We strive to create an environment of inclusion—one focused on psychological safety, empathy, and human connection—that will allow employees of all backgrounds to thrive.

About The Role & Team

The Workspaces team is a core vertical product team responsible for driving engagement and retention of our core Analytics product. We’ve been industry leaders in the data visualization space and have first in class tools in behavioral analytics. We are taking the next step in this journey by automating insights and workflows that are performed manually by our customers today.

 As a Staff ML Engineer, Applied AI, you will: 

• Partner closely with PM and Design, and AI Infra teams to build workflows, automations, and insights that leverage our existing products.
• Collaborate with AI Infra teams to both define and expand existing capabilities
• Build ML and LLM systems to surface anomalies, insights, and suggested actions based on large amounts of event data
• Integrate with LLMs and experiment with different methodologies of providing context to generate relevant and actionable insights
• Launch and drive adoption of automated insights

You'll be a great addition to the team if you have:

• You have 5+ years of experience working with ML systems and pipelines
• You have experience building and deploying services on AWS or similar cloud providers
• You have proficiency in at least one of the following: Python, Javascript
• You have experience with both relational and NoSQL databases
• You have experience shipping consumer facing products leveraging LLMs
• You believe in shipping quickly and learning through experimentation

Who We Are

The Company: Amplitude is filled with humble, life-long learners who are eager to help one another and the company succeed. Our values of growth mindset, ownership, and humility are core to the way we work: we’re tenacious in the face of challenges, we take the initiative to solve problems that drive our shared success, and we operate from a place of empathy and openness, seeking to understand many points of view. 

The Product: Amplitude is a digital analytics platform—we help companies capture data they can trust, uncover clear insights about customer behavior, and take faster action. This empowers teams to build better product experiences that drive business growth. We’re super proud of what we’ve built and continue to expand: a platform that empowers companies to thrive in the digital era.

We care about the well-being of our team: We offer competitive pay and benefits packages that reflect our commitment to the health and well-being of our Ampliteers.

Some of our benefit programs include:

• Excellent ​M​edical, ​D​ental and ​V​ision insurance coverages, with 100% employer-paid premiums for employee ​M​edical, ​D​ental,​ ​​​​​​​​Vision on select plans
• Flexible time off, ​p​aid holidays, and more
• Generous stipends to spend on what matters most to you, whether that’s wellness (monthly), commuter transit/parking (monthly), learning and development (quarterly), home office equipment (annual), and much more
• Excellent Parental benefits including​:​ 12-20 weeks of Paid Par