Software Development Engineer, AI/ML, AWS Neuron, Model Inference

What you’d do

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