Software Engineer II - AI/ML, AWS Neuron

What you’d do

Build and optimize distributed inference support for PyTorch in the Neuron SDK, tuning models 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.