Showing 71,287 verified jobs

Software Engineer, AI/ML, AWS Neuron
View job details and apply for this position.
Amazon Logo
AM

Software Engineer, AI/ML, AWS Neuron

Amazon

Early applicant

$129K – $224K/yr

On-site

Employee

Full-Time

Mid Level

AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators and the Trn1 and Inf1 servers that use them. This role is for a software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. This role is responsible for development, enablement and performance tuning of a wide variety of ML model families, including massive scale large language models like LLama4, Mixtral, DBRX and beyond, as well as stable diffusion,...

Skills

Software development

Machine learning

Python

Pytorch

Jax

XLA

Neuron compiler

TensorFlow

Deepspeed

FSDP