πŸ“ Jobs Near Me
πŸ“

HiringNearMe.work

Local Jobs, Zero Commute

πŸ“ Local Job Near You

Neuron Collectives Software Engineer, Trainium Collectives

🏒
Amazon
πŸ“ Cupertino, United States
πŸ“
Location Cupertino
πŸ“…
Posted June 06, 2026
πŸš—
Commute Local Area
🎯
Local Opportunity Near You!

This job is in your area. Enjoy a short commute and work close to home.

πŸ“‹
Job Description

Description
As a Neuron Collectives Software Developer, you will:

* Enhance collective algorithms and topologies for optimal training performance
* Use tools like Neuron Explorer to identify bottlenecks in compute and bus bandwidth utilization
* Monitor and analyze processor, DMA, firmware, and workload metrics
* Optimize collective operations to scale AI compute across the data center through low level device driver development
* Work closely with the hardware team to co-optimize software and Trainium silicon
* Develop and optimize C/C++ implementations of collective communication patterns
* Investigate and implement improvements for specific training topologies used by modern LLMs
* Build and maintain analysis frameworks and automation solutions

The role offers opportunities to work on cutting-edge AI training hardware while contributing to one of Amazon's most critical initiatives.

A day in the life
Annapurna Labs, a crucial part...

Apply for This Job

Submit Application

Quick and secure application process

πŸ“ Location Details

πŸŒ†
City
Cupertino
πŸ—ΊοΈ
Country
United States
πŸš—
Commute
Local Area

πŸ” More Jobs Nearby

Explore other opportunities in Cupertino

View Local Jobs