Staff Ml Engineer Trainium Sagemaker

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:
Strong 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