Machine Learning Engineer - HSTU
from
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.
Key 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
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.