Staff ML Engineer – AWS Trainium & SageMaker
Core
Operate and train large-scale AI models on Amazon SageMaker using AWS Trainium custom silicon, bridging hardware-level compilation with production PyTorch pipelines.
Role type
Staff ML Engineer (AWS Trainium & SageMaker)
Builds
Production training pipelines and optimized model training workloads for enterprise clients
Domain
Cloud AI / Custom Silicon Accelerators
Deliverable
production ML models
Required skills
PyTorch (distributed/multi-device), Amazon SageMaker, Python, hardware-level debugging, cost-aware pipeline design
Preferred skills
AWS Trainium/Inferentia (Neuron SDK), NeuronCore architecture, compiler behavior understanding
Technologies
PyTorch, Amazon SageMaker, AWS Trainium, Neuron SDK, Python
