This role is for a software engineer in the Compiler team for AWS Neuron, focusing on building the next generation Neuron compiler to optimize ML models for AWS Inferentia and Trainium. The team values knowledge-sharing and mentorship, fostering an inclusive culture that supports diverse experiences. You will work closely with chip architects and ML applications teams to enhance compiler performance and user experience. The position offers opportunities to work with open-source software and contribute to innovative features for developers globally.

• 3+ years of non-internship professional software development experience
• 2+ years of non-internship design or architecture experience of new and existing systems
• Experience programming with at least one software programming language
• Master's degree or PhD in Computer Science or a related technical field
• 3+ years of experience writing production grade code in object-oriented languages such as C++ or Java
• Experience in compiler design for CPU, GPU, Vector engines, or ML-accelerators
• Experience with OpenSource compiler toolset like LLVM or MLIR
• Experience with technologies such as PyTorch, OpenXLA, StableHLO, JAX, or TVM
• Experience with modern build systems like Bazel or CMake

• Design, implement, test, deploy, and maintain innovative software solutions for the Neuron compiler
• Collaborate with chip architects, runtime engineers, and ML applications teams to deploy ML models on AWS accelerators
• Create compiler optimization and verification passes to enhance performance and stability
• Build features that surface the peculiarities of AWS accelerators to developers
• Implement tools to analyze numerical errors and resolve compiler defects
• Participate in design discussions and code reviews with internal and external stakeholders

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