Staff ML Engineer
Get new jobs like this one by email. Create a free account for job alerts and to save the jobs you like.
Job Summary
To support enterprise clients with production AI systems, the remote Staff ML Engineer will train and operate models on Amazon SageMaker using AWS Trainium, focusing on optimizing PyTorch training code and diagnosing hardware-specific issues.
Key responsibilities
- Train and operate models on Amazon SageMaker with AWS Trainium as the underlying compute
- Write and optimize PyTorch training code with an understanding of its compilation and execution on Trainium
- Diagnose training run issues related to hardware and translate requests into cost-aware training pipelines
Required qualifications
- Strong, hands-on experience with PyTorch, including distributed or multi-device training
- Production experience with Amazon SageMaker for training and/or inference
- Understanding of device-specific compilation and debugging hardware-related issues
- AWS Trainium or Inferentia experience is a strong plus
- Solid Python fundamentals and experience in a client-facing, production engineering environment
Complete Job Description
The complete job description is available to premium members. Premium membership includes:
Full access to 38,022 remote jobs from human-vetted companies, updated daily
Resume Builder - AI-powered tool to craft, enhance, and tailor your resume to a specific job
Twice-monthly live group coaching and the full Remote Career Center
20% member discount on Career Services
Backed by a 30-day money-back guarantee
Company Company Name
Headquarters Headquarters
Founded Founded
Website
The company description is available to premium members. Go Premium for full access to every job and company profile in our remote jobs database, backed by a 30-day money-back guarantee.
