Staff ML Engineer building production PyTorch training pipelines on AWS Trainium and SageMaker. Delivering cost-aware, hardware-optimized AI systems for enterprise clients.
Responsibilities
Train and operate models on Amazon SageMaker with AWS Trainium as the underlying compute
Write and optimize PyTorch training code for Trainium, including reasoning about NeuronCore architecture, compiler behavior, memory, and throughput tradeoffs
Diagnose hardware-specific training run issues and distinguish data or code problems from compiler- or device-level problems
Translate Trainium job requests into working, cost-aware end-to-end training pipelines
Tune distributed training runs for throughput and cost on SageMaker training infrastructure
Work directly with client and internal engineering teams to scope and deliver production training workloads
Requirements
Strong, hands-on PyTorch experience, ideally including distributed or multi-device training
Production experience with Amazon SageMaker for training and/or inference
Comfort working close to the hardware layer, including device-specific compilation and accelerator-level debugging
AWS Trainium or Inferentia (Neuron SDK) experience is a strong plus
Deep PyTorch experience and a track record of picking up new hardware targets quickly may substitute for Trainium experience
Solid Python fundamentals
Comfort operating in a client-facing, production engineering environment
Cloud Engineer designing and operating secure AKS platforms for Manulife’s Canadian financial services business. Automating infrastructure, improving reliability, and integrating developer platform capabilities.
Cloud Platform Engineer managing Kubernetes and Docker infrastructure for EQ Bank, Canada’s challenger bank. Supporting cloud operations, automation, monitoring, disaster recovery, and production incidents.
Architecte infonuagique chez TEHORA, firme québécoise de services techniques et de gestion de projets. Conception et gouvernance de plateformes cloud, sécurité, conformité et accompagnement des équipes.
Staff Cloud Architect leading cloud, data, and AI/ML architecture for Robots and Pencils’ applied AI engineering engagements. Designing scalable, secure production systems across US and Canadian remote teams.
Senior Cloud Architect designing cloud - native, data, and AI/ML systems for Robots & Pencils, an applied AI engineering firm. Building production - ready enterprise solutions across US and Canada.
Student Cloud Infrastructure Analyst at Sun Life, helping operate enterprise AI and AWS cloud platforms. Supporting Infrastructure as Code, CI/CD, automation, and DevOps initiatives.
Cloud infrastructure manager leading Azure operations, migrations, governance, and team delivery. Supporting Coastal Community Credit Union’s financial services and local member communities.
Cloud Engineer supporting Sun Life’s financial security and wellness technology infrastructure. Assisting cloud migration, automation, security, and multi - cloud initiatives across hybrid environments.
Data Developer building Azure data pipelines and architectures. Transforming enterprise data with Azure Synapse, Data Factory, Spark, Python, and Power BI.