AI/ML Performance Engineer – Training Trainium
Listed on 2026-10-06
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer, AWS
Amazon Web Services (AWS) is seeking a software professional to lead optimization of distributed training on AWS Trainium, aiming to maximize throughput and model FLOPs utilization. You will diagnose bottlenecks across the stack, from collective communications to compiler optimizations, and drive end-to-end improvements.
Requirements include 3+ years of software development experience and 2+ years in system design or architecture, with proficiency in at least one programming language.
Consider building your career as a AI/ML Performance Engineer – Training on Trainium at Amazon Web Services (AWS).
Join us at Amazon Web Services (AWS) as our next AI/ML Performance Engineer – Training on Trainium in Cupertino, CA, United States.
We are currently recruiting a AI/ML Performance Engineer – Training on Trainium for our team in Cupertino, CA, United States.
This is a great opportunity to take on the AI/ML Performance Engineer – Training on Trainium role at Amazon Web Services (AWS).
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).