Sr. Software Engineer- AI/ML, AWS Neuron

What you’d do

Build and optimize distributed inference support for PyTorch in the Neuron SDK, tuning ML models like GPT, Kimi, and Qwen to run efficiently on AWS Trainium and Inferentia hardware. Full-time role collaborating across compiler, runtime, framework, and hardware teams. Found on the employer's careers page on Sep 30, 2026. Apply through the link.