Robots and Pencils - Staff ML Engineer – AWS Trainium & SageMaker

• Train and operate models on Amazon SageMaker with AWS Trainium as the underlying compute • Write and optimize PyTorch training code with a real understanding of how it compiles and executes on Trainium (NeuronCore architecture, compiler behavior, memory and throughput tradeoffs) • Diagnose training run issues that show up specifically because of the hardware, not just the model, distinguishing a data or code problem from a compiler or device-level one • Translate a request for "a Trainium job" into an actual working, cost-aware training pipeline, end to end • Tune distributed training runs for throughput and cost on SageMaker's training infrastructure • Work directly with client and internal engineering teams to scope and deliver real production training workloads, not experiments that stay in a notebook