The Software Development Engineer role at Annapurna Labs focuses on building and evolving machine learning tools for custom AI accelerators. The team is high-impact and high-visibility, directly enhancing the capabilities of over 100 engineers. This position offers significant design ownership and the opportunity to drive architectural decisions in a startup-like environment. The work involves collaboration with hardware engineers, system architects, and ML researchers to optimize ML workloads.

• Bachelor's degree or above in computer science or equivalent.
• Experience in Kubernetes, Docker, or the containers ecosystem.
• Proficiency in Java and at least one of Go, Python, or TypeScript.
• Familiarity with Git and CI/CD pipelines.
• Internship or project experience with AWS services such as EKS, EC2, or S3.
• Experience with Linux systems and performance profiling.

• Design and implement tooling for profiling, optimization, and resource management of ML workloads.
• Build high-impact solutions that ship to a large and growing customer base.
• Participate in design discussions, code reviews, and cross-functional collaboration.
• Create metrics, implement automation, and resolve root causes of software defects.
• Work in a startup-like environment focused on solving important problems.

• Employees at Amazon are often offered comprehensive health benefits—including multiple medical plan options (no pre-existing condition exclusions, 100% covered in-network preventive care), dental and vision plans, a 24/7 medical advice line from day one, expert second-opinion services, and broad mental-health support with several free counseling sessions (including pediatric). Financial wellness typically includes a 401(k) with company match (up to 2%), Restricted Stock Units (equity), FSAs, an emergency savings program, product and partner discounts, and even college-savings and home-purchase programs. Overall, the package is designed to support employees and their families’ health, finances, and day-to-day life.