(Fall 2026) Annapurna Labs at AWS Internship (US) - Machine Learning Systems & Silicon Innovation

About the role

At Annapurna Labs, an Amazon company, we're not just participating in the AI revolution, we're accelerating it. We design custom silicon and revolutionary software systems that were considered impossible just yesterday, powering the world's most advanced AI infrastructure at AWS. From training the largest language models on Earth to developing breakthrough ML accelerator chips, we're inventing the future of cloud computing.

Technical Tracks

• Framework Optimization (PyTorch, JAX)
• Compiler Development & Optimization
• Distributed Training Systems
• Performance Engineering
• ML Infrastructure Development
• Firmware & Driver Development
• Runtime Systems
• Fleet Management & Automation
• Hardware/Software Integration
• Performance Analysis Tools
• RTL Development for ML Accelerators
• Hardware Architecture & Modeling
• Physical Design & Power Optimization
• Custom Circuit Design
• Pre/Post Silicon Validation

Basic Qualifications

• Work 40 hours/week minimum and commit to 12 week internship maximum
• Currently pursuing a BS, MS, or PhD in Computer Science, Computer Engineering, Electrical Engineering, or related technical field
• Must be enrolled in a full-time degree program at time of application and returning to school after the internship
• Programming experience in C, C++, Python, or similar languages
• Experience in two or more of the following areas: 1. Systems programming or low-level software development 2. Compiler design or optimization 3. Machine learning frameworks (PyTorch, JAX, TensorFlow) 4. Distributed systems or parallel computing 5. Performance analysis and optimization 6. Hardware design (RTL, Verilog, FPGA development)

Preferred Qualifications

• Previous internship, research, or project experience in hardware/software co-design, ML systems, or computer architecture
• Contributions to open-source projects or research publications
• Completed or currently enrolled in coursework covering machine learning, parallel computing, computer architecture and/or compiler construction

Benefits

The starting pay for this position is listed below. Final starting pay will be based on factors including experience, qualifications, and location.

• USA, CA, Cupertino - 127,100.00 - 185,000.00 USD annually
• USA, TX, AUSTIN - 110,500.00 - 160,000.00 USD annually
• USA, TX, Austin - 110,500.00 - 160,000.00 USD annually
• USA, WA, SEATTLE - 110,500.00 - 160,000.00 USD annually
• USA, WA, Seattle - 110,500.00 - 160,000.00 USD annually