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, providing a supportive environment for career growth. You will work closely with chip architects and ML applications teams to enhance compiler performance and user experience. The position emphasizes collaboration and innovation in a dynamic, startup-like environment.

• 3+ years of non-internship professional software development experience
• 2+ years of non-internship design or architecture experience
• 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 toolsets like LLVM or MLIR
• Experience with technologies such as PyTorch, OpenXLA, StableHLO, JAX, or TVM

• Design, implement, test, deploy, and maintain innovative software solutions for the Neuron compiler.
• Solve compiler optimization problems to achieve optimum performance for various ML model families.
• Collaborate with chip architects, runtime engineers, and ML applications teams to deploy ML models on AWS accelerators.
• Create compiler optimization and verification passes and implement tools to analyze numerical errors.
• Participate in design discussions and code reviews, communicating 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.