Save application

Annapurna Labs (U.S.) Inc.

Manual apply

1 hour ago

Software Development Engineer, Neuron Collectives, Annapurna Labs

Onsite · US, CA, Cupertino United States

aws software development

amazon

Match & cover letter

Create a profile to see your match and get a cover letter for this role.

Create profile and match

Source: amazon

Never pay or share passwords or codes with an employer. How to spot a fake job

AI TL;DR

Join the AWS Neuron Collectives team to optimize AI training operations for Trainium. You'll enhance algorithms and work closely with hardware to maximize performance in a collaborative environment.

AI match

Clear role · 80/100 against the market band on this page.

Interview prep for this job

  1. What would you ship in the first month as Software Development Engineer, Neuron Collectives, Annapurna Labs at Annapurna Labs (U.S.) Inc.?
  2. Which tradeoff in the description would you ask about first?
  3. How do you explain a production incident to this team?
  4. What would you measure in the first two weeks?
  5. Which part of your last project maps to this stack?

Five questions are free. The rest are Pro. See Pro

Similar roles

Job description

Join the AWS Neuron Collectives team to optimize AI training operations for Trainium. You'll enhance algorithms and work closely with hardware to maximize performance in a collaborative environment.

Details

  • Cupertino, CA, onsite
  • Experience with AWS and C/C++ required

The work

  • Enhance collective algorithms for optimal training performance
  • Use Neuron Explorer to identify compute bottlenecks
  • Monitor processor, DMA, firmware, and workload metrics
  • Optimize collective operations to scale AI compute