ZipRecruiter

Log In

1

Aws Neuron Jobs (NOW HIRING)

Amazon

Sr Solutions Architect, Annapurna ML

Cupertino, CA · On-site

$80 - $104.75/hr

AWS Nitro, ENA, EFA, Graviton and F1 EC2 Instances, AWS Neuron, Inferentia and Trainium ML Accelerators, and in storage with scalable NVMe, are some of the products we have delivered, over the last ...

Amazon

Sr Solutions Architect, Annapurna ML

Cupertino, CA · On-site

$80 - $104.75/hr

AWS Nitro, ENA, EFA, Graviton and F1 EC2 Instances, AWS Neuron, Inferentia and Trainium ML Accelerators, and in storage with scalable NVMe, are some of the products we have delivered, over the last ...

Amazon

Software Engineer II- AI/ML, AWS Neuron

Seattle, WA

$111K - $151K/yr

The AWS Neuron SDK, developed by the Annapurna Labs team at AWS, is the backbone for accelerating deep learning and GenAI workloads on Amazon's Trainium ML accelerators. This comprehensive toolkit ...

next page

Showing results 1-20

All JobsAws Neuron Jobs

Aws Neuron information

See salary details

$38

$60

$89

How much do aws neuron jobs pay per hour?

As of Jun 22, 2026, the average hourly pay for aws neuron in the United States is $60.47, according to ZipRecruiter salary data. Most workers in this role earn between $48.56 and $69.71 per hour, depending on experience, location, and employer.

Is ML a high paying job?

Machine Learning (ML) roles, including positions like AWS Neuron developers, tend to offer high salaries due to the specialized skills required, such as programming, data analysis, and knowledge of AI frameworks. Compensation varies based on experience, location, and industry, but generally, ML jobs are considered well-paying compared to many other tech roles.

What jobs can you do with neural engineering?

Neural engineering offers roles such as neural engineer, biomedical engineer, research scientist, and neurotechnology developer. These jobs involve designing brain-computer interfaces, developing neural prosthetics, and conducting research using tools like neuroimaging and signal processing, often requiring knowledge of neuroscience, engineering, and programming. Certification and advanced degrees can enhance job prospects in this field.

What AWS jobs are in high demand?

AWS roles such as Cloud Engineer, Solutions Architect, DevOps Engineer, and Cloud Security Specialist are in high demand due to the widespread adoption of cloud services. These positions often require skills in cloud infrastructure, scripting, and certifications like AWS Certified Solutions Architect or AWS Certified DevOps Engineer.

What does AWS Neuron do?

AWS Neuron is a software development framework that enables machine learning models to run efficiently on AWS Inferentia and Trainium chips. It provides optimized libraries and tools for deploying deep learning workloads, often requiring knowledge of AWS cloud services and neural network frameworks. The role of an AWS Neuron specialist involves configuring, optimizing, and maintaining these AI inference and training environments.
More about Aws Neuron jobs
What cities are hiring for Aws Neuron jobs? Cities with the most Aws Neuron job openings:
What states have the most Aws Neuron jobs? States with the most job openings for Aws Neuron jobs include:
What job categories do people searching Aws Neuron jobs look for? The top searched job categories for Aws Neuron jobs are:
Aws Neuron jobs near you
Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

Amazon

Cupertino, CA

$151K - $199K/yr

Full-time

Posted 21 days ago

Amazon rating

7.4

Company rating: 7.4 out of 10

Based on 6,873 frontline employees who took The Breakroom Quiz

6th of 39 rated national retailers

Job description

Annapurna Labs designs silicon and software that accelerates innovation. Our custom chips, accelerators, and software stacks enable us to tackle unprecedented technical challenges and deliver solutions that help customers change the world. AWS Neuron is the complete software stack powering AWS Trainium (Trn2/Trn3), our cloud scale Machine Learning accelerators and we are seeking a Senior Software Engineer to join our ML Distributed Training team.
In this role, you will be responsible for the development, enablement, and performance optimization of large scale ML model training across diverse model families

This includes massive scale pre-training and post-training of LLMs with Dense and Mixture-of-Experts architectures, Multimodal models that are transformer and diffusion based, and Reinforcement Learning workloads. You will work at the intersection of ML research and high performance systems, collaborating closely with chip architects, compiler engineers, runtime engineers and AWS solution architects to deliver cost-effective, performant machine learning solutions on AWS Trainium based systems.
Key job responsibilities
You will design, implement and optimize distributed training solutions for large scale ML models running on Trainium instances. A significant part of your work will involve extending and optimizing popular distributed training frameworks including FSDP (Fully-Sharded Data Parallel), torchtitan and Hugging Face libraries for the Neuron ecosystem.
A core focus of this role involves developing and optimizing mixed-precision and low-precision training techniques

You will work with BF16, FP8, and emerging numerical formats to maximize training throughput while maintaining model accuracy and convergence quality. This requires implementing precision aware training strategies, loss scaling techniques, and careful gradient management to ensure training stability across reduced precision formats. Understanding the tradeoffs between computational efficiency and numerical fidelity is essential to success in this position.
Beyond precision optimization, you will profile, analyze, and tune end-to-end training pipelines to achieve optimal performance on Trainium hardware

You will partner with hardware, compiler, and runtime teams to influence system design and unlock new capabilities. Additionally, you will work directly with AWS solution architects and customers to deploy and optimize training workloads at scale.
About the team
Annapurna Labs was a startup company acquired by AWS in 2015, and is now fully integrated. If AWS is an infrastructure company, then think Annapurna Labs as the infrastructure provider of AWS

Our org covers multiple disciplines including silicon engineering, hardware design and verification, software, and operations. AWS Nitro, ENA, EFA, Graviton and F1 EC2 Instances, AWS Neuron, Inferentia and Trainium ML Accelerators, and in storage with scalable NVMe, are some of the products we have delivered, over the last few years.
.

What Amazon employees say

Pay

Benefits

Hours and flexibility

Workplace

Get the full story on Breakroom

Amazon logo

About Amazon

Sourced by ZipRecruiter

Amazon.com, Inc., commonly known as Amazon, is an American multinational technology company. It was founded by Jeff Bezos in 1994 and initially started as an online marketplace for books. Since then, Amazon has expanded its operations and become one of the largest e-commerce companies in the world. Amazon's primary business is its online retail platform, where customers can purchase a vast array of products, including electronics, clothing, books, home goods, and much more. The company offers a convenient and user-friendly shopping experience, with features such as fast shipping, customer reviews, and personalized recommendations. In addition to its e-commerce platform, Amazon has diversified its business into various other areas. One of its notable ventures is Amazon Web Services (AWS), a comprehensive cloud computing platform that provides services such as storage, compute power, and database management to individuals and businesses. AWS has become a leader in the cloud computing industry, powering many websites and applications worldwide. Amazon has also developed its own consumer electronics, including the popular Amazon Kindle e-reader, Fire tablets, Fire TV streaming devices, and the Alexa-powered Echo smart speakers. The Alexa voice assistant, integrated into these devices, allows users to interact with their devices using voice commands, perform tasks, and access information. Furthermore, Amazon has expanded into media and entertainment. It operates Prime Video, a streaming service that offers a wide range of movies, TV shows, and original content. Amazon Music provides a platform for streaming and purchasing digital music, while Audible offers audiobooks and other audio content. The company's commitment to customer satisfaction and convenience is demonstrated by its membership program, Amazon Prime. Prime members receive various benefits, including free two-day shipping, access to streaming services, exclusive deals, and more.

Industry

It services, book publishers, retail, real estate and computer and electronic product manufacturing

Company size

10,000+ Employees

Headquarters location

Seattle, WA, US

View All Amazon Jobs

Apply