Software Engineer- AI/ML, Amazon Neuron Training

What you’d do

Implement and tune components of a distributed training stack for large-scale training, post-training, and reinforcement learning on Trainium instances, working across PyTorch and the Neuron software stack. Full-time role profiling workloads, applying parallelism strategies, and building internal tooling, benchmarks, and tests. Found on the employer's careers page on Oct 6, 2026. Apply through the link.

You’ll need

• Bachelor's degree or above in computer science or equivalent
• 2+ years designing or architecting new and existing systems
• 3+ years of full software development life cycle in production
• PyTorch
• Distributed training and parallelism strategies