Build production AI systems remotely from Canada using deep PyTorch expertise across AWS Trainium and Amazon SageMaker training pipelines.
Robots & Pencils seeks a Staff ML Engineer to operate and train models on Trainium-powered SageMaker infrastructure. You will reason across PyTorch code, NeuronCore architecture, compiler behavior, memory, and throughput. The work supports enterprise clients through forward-deployed engineering teams delivering production training workloads.

Key Responsibilities

Train and operate models on Amazon SageMaker with AWS TrainiumWrite and optimize PyTorch training code for Trainium executionDiagnose hardware, compiler, device, data,



and code issuesTranslate Trainium requests into cost-aware production pipelinesTune distributed training for SageMaker throughput and costRequirements:
Solid hands-on PyTorch experience, ideally distributed trainingProduction experience with Amazon SageMaker training or inferenceUnderstanding of device-specific compilation and accelerator debuggingAWS Trainium or Inferentia Neuron SDK experience is valuableSolid Python fundamentals and client-facing production engineering comfortApply deep PyTorch, SageMaker, and accelerator expertise to production AI workloads on AWS Trainium for enterprise clients.
#J-18808-Ljbffr

📌 Staff Ml Engineer Trainium Sagemaker (Winnipeg)
🏢 Robots and Pencils
📍 Winnipeg