Shopify is hiring a Machine Learning Engineer to build and deploy HSTU (Hierarchical Sequential Transduction Unit) recommendation models that surface growth drivers for merchants and personalization for over 100 million shoppers. You’ll work on high-impact AI problems across Generative AI, NLP, large-scale inference, data pipelines, and distributed GPU training. The role may require on-call work and is remote by design.
Overview
$194,000 – $242,000 / year
remote
full-time
No experience required (senior level)
Technologies
- Generative AINLPHSTURecommendation SystemsLLMsPythonShell ScriptingStreaming and Batch Data PipelinesVector DatabasesDBTBigQueryBigTableOrchestration ToolsDistributed GPU TrainingANN SearchNegative Sampling
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 including merchant events, consumer interactions, and payment sequences
- Experiment with novel architectures
- Optimize for production through advanced techniques such as 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, Generative AI, or LLMs
- End-to-end experience training, evaluating, testing, and deploying machine learning products at scale
- Experience 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 running machine learning in parallel environments, such as distributed clusters or GPU optimization
Benefits
- Comprehensive medical, dental, and vision coverage
- Wellness programs
- Generous parental leave policy
- Fertility and family-building benefits
- Flexible work, remote by design
- Home office setup and support
- Destination 90: work from another destination for up to 90 days per calendar year
- Yearly travel budget to visit a Shopify office or travel
- Flex Comp: choose cash, restricted share units (RSUs), or both