Machine Learning Engineer - HSTU

Role Overview

Discipline / Category: Applied ML (Applied ML Engineering)

Shopify is hiring for a Machine Learning Engineer - HSTU. Shopify is a digital by default company where employees work remotely across the Americas.

Join Shopify's innovative team as we work on the development and implementation of state of the art HSTU models (Hierarchical Sequential Transduction Unit) to recommend the best growth drivers and action for merchants and buyers. You'll play a pivotal role in solving high-impact data problems that directly improve merchant success and consumer experience.

As a Machine Learning Engineering (MLE) lead or individual contributor, you'll be at the forefront of building AI solutions that anticipate both merchant needs and personalization for 100M+ shoppers.

Responsibilities

• Develop and deploy Generative AI, natural language processing, and HSTU-based recommendation models at scale

• Design and implement scalable AI/ML system architectures supporting models

• Build sophisticated inference pipelines that process billions of events and deliver real-time recommendations

• Implement data pipelines for model training, fine-tuning, and evaluation across diverse data sources (merchant events, consumer interactions, payment sequences)

• Experiment with novel architectures

• Optimize for production through advanced techniques like negative sampling, ANN search, and distributed GPU training

• Collaborate cross-functionally with product teams, data scientists, and infrastructure engineers to deliver measurable business impact

• Communicate effectively with both technical and non-technical audiences, translating complex ML concepts into actionable insights

Requirements & Qualifications

• Mastery in recommendation systems, Gen AI or LLMs

• End-to-end experience in training, evaluating, testing, and deploying machine learning products at scale.

• Experience in building data pipelines and driving ETL design decisions using disparate data sources.

• Proficiency in Python, shell scripting, streaming and batch data pipelines, vector databases, DBT, BigQuery, BigTable, or equivalent, and orchestration tools.

• Experience with running machine learning in parallel environments (e.g., distributed clusters, GPU optimization).

• This role may require on-call work.

At Shopify, we pride ourselves on moving quickly—not just in shipping, but in our hiring process as well. If you’re ready to apply, please be prepared to interview with us within the week. Our goal is to complete the entire interview loop within 30 days. You will be expected to complete a pair programming interview, using your own IDE.This role may require on-call work.Ready to redefine e-commerce through AI innovation? Join the team that’s making commerce better for everyone.

Total Rewards & Flex Comp

Shopify offers a flexible compensation system (Flex Comp) that allows employees to choose their own balance between cash (base salary) and equity (RSUs and stock options). Work is digital by default with remote flexibility across the Americas.