Machine Learning Engineer - HSTU

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.