This role involves joining a dynamic team focused on leveraging Generative AI to enhance customer adoption of Amazon's Neuron software stack. The team aims to simplify and accelerate the migration to AWS's Machine Learning silicon, particularly Trainium. As a key contributor, you will work on building AI agents and tools that drive impact in Amazon's Generative AI business. The position is based in New York City and is part of Amazon Annapurna Labs, which is dedicated to innovation in silicon and software for machine learning.

• 3+ years of relevant, broad engineering experience.
• Bachelor's or Master's degree in computer science or equivalent.
• Hands-on technical experience working in the Generative AI space.
• Experience in one or more of the following areas: ML compilers, production coding agents, GenAI model architecture, model training, neural network optimization, or alternatively applied math.
• Domain-level knowledge of AWS services.

• Research implementations that deliver the best possible experiences for customers.
• Deliver on goals to improve the time and effort it takes to port and optimize Machine Learning workloads on Neuron.
• Build and operate agentic evaluation frameworks, monitoring, and feedback loops that ensure the correctness, reliability, and continuous improvement of AI agents.
• Design, implement, test, deploy and maintain innovative software solutions to transform service performance, durability, cost, and security.
• Build high-quality, highly available, always-on products.

• 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.