Software Development Engineer I, ML Infra Services, Annapurna Labs

Amazon's Annapurna Labs is seeking a Software Development Engineer to enhance machine learning tools for optimizing workloads on custom AI accelerators. The role involves collaboration with hardware engineers and ML researchers, focusing on building impactful solutions and participating in design discussions.

Responsibilities:

• Design and implement tooling for profiling, optimization, and resource management of ML workloads on custom accelerators
• Build high-impact solutions that ship to a large and growing customer base
• Participate in design discussions, code reviews, and cross-functional collaboration with hardware, software, and customer-facing teams
• Create metrics, implement automation, and resolve root causes of software defects
• Work in a startup-like environment where you're always focused on the most important problems