(fall 2026) Annapurna Labs at Aws Internship (us) - Machine Learning Systems & Silicon Innovation

Amazon Amazon · Big Tech · Cupertino, CA · Software Development

Internship role focused on building AI infrastructure and ML systems, including custom silicon design, framework optimization, compiler development, and distributed training systems for AWS.

What you'd actually do

  1. Framework Optimization (PyTorch, JAX)
  2. Compiler Development & Optimization
  3. Distributed Training Systems
  4. Performance Engineering
  5. ML Infrastructure Development

Skills

Required

  • C
  • C++
  • Python
  • Systems programming
  • low-level software development
  • Compiler design
  • optimization
  • Machine learning frameworks
  • Distributed systems
  • parallel computing
  • Performance analysis
  • optimization
  • Hardware design
  • RTL development
  • FPGA development

Nice to have

  • hardware/software co-design
  • ML systems
  • computer architecture
  • open-source projects
  • research publications
  • machine learning coursework
  • parallel computing coursework
  • computer architecture coursework
  • compiler construction coursework

What the JD emphasized

  • custom silicon
  • revolutionary software systems
  • ML accelerator chips
  • AI infrastructure
  • AWS
  • training the largest language models on Earth
  • breakthrough ML accelerator chips
  • inventing the future of cloud computing
  • accelerating AI innovation
  • AI acceleration

Other signals

  • custom silicon
  • ML accelerator chips
  • AI infrastructure
  • distributed training systems
  • ML infrastructure development