The Software Development Engineer role at Annapurna Labs focuses on building and evolving machine learning tools that optimize and analyze workloads on custom AI accelerators. The team operates in a high-impact environment, directly enhancing the capabilities of over 100 engineers. This position involves significant design ownership and collaboration with hardware engineers, system architects, and ML researchers. Candidates will work at the forefront of AI infrastructure, engaging with cutting-edge technologies.

• Candidates must be 18 years of age or older.
• 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.
• Familiarity with Git and CI/CD pipelines.
• Experience from a technical internship.
• Experience in optimization mathematics such as linear programming and nonlinear optimization.
• Experience with distributed, multi-tiered systems, algorithms, and relational databases.
• Experience with Cloud platforms, preferably AWS.
• Familiarity 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 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 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.