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

• Experience with at least one modern programming language such as Java, Python, C++, or C#.
• Proficiency in Java and at least one of Go, Python, or TypeScript.
• Experience with data structure implementation and basic algorithm development.
• Familiarity with Git and CI/CD pipelines.
• Experience with Cloud platforms, preferably AWS.
• Experience with distributed systems or big data architectures.
• Experience with Linux systems and performance profiling.

• Design and implement tooling for profiling, optimization, and resource management of ML workloads.
• Build high-impact solutions for a large and growing customer base.
• Participate in design discussions and code reviews.
• Create metrics, implement automation, and resolve software defects.
• Work in a startup-like environment focused on 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.