Staff ML Engineer – AWS Trainium & SageMaker
You'll train and optimize large-scale models on AWS SageMaker using Trainium custom silicon, writing PyTorch code with deep understanding of how it compiles and executes on the und...