Lead ML Systems Engineer – AI/GenAI Acceleration
Listed on 2026-10-06
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer
Amazon Web Services (AWS) Annapurna Labs seeks a technical leader to build and optimize distributed training solutions for Trainium accelerators, integrating with PyTorch and JAX. You will own parallelism strategies and work across compiler, runtime, and frameworks to push frontier-scale models.
Join a team that values innovation in ML acceleration, high-performance computing, and scalable systems design. The role emphasizes collaboration, mentoring, and delivering measurable performance
This is a genuine position to take on the Lead ML Systems Engineer – AI/GenAI Acceleration role at Amazon Web Services (AWS).
We would love to welcome a new Lead ML Systems Engineer – AI/GenAI Acceleration to our group in Cupertino, CA, United States.
For the Lead ML Systems Engineer – AI/GenAI Acceleration position at Amazon Web Services (AWS), we are reviewing applications now.
(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).