Join Robots & Pencils as an AWS Trainium ML Engineer focusing on model training with Amazon SageMaker. This role demands solid PyTorch skills and understanding of custom hardware. Robots & Pencils seeks an experienced engineer to train and optimize models using Amazon SageMaker and AWS Trainium.

You'll work embedded with client teams, diving deep into compilation and execution behaviors at both the hardware and software levels. Your expertise in PyTorch will be essential, especially for distributed and multi-device training. Key Responsibilities:

- Train models on Amazon SageMaker using AWS Trainium
- Optimize PyTorch code based on custom silicon compilation
- Diagnose hardware-related training run issues
- Develop cost-effective training pipelines from requests
- Tune distributed training for efficiency on SageMaker Requirements:
- Hands-on experience with PyTorch, especially in distributed settings
- Production experience using Amazon SageMaker
- Ability to troubleshoot hardware-specific issues
- Familiarity with AWS Trainium or similar
- Strong Python skills and client-facing experience Utilize your advanced PyTorch skills and production experience to create impactful AI systems at Robots & Pencils.

📌 AWS Trainium ML Engineer at Robots & Pencils (Montreal)
🏢 Robots-u0026-Pencils-
📍 Montreal