This role is for a senior 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 mentorship and knowledge-sharing, creating an inclusive environment for diverse experiences. You will work closely with chip architects and ML application teams to enhance compiler performance and user experience. The position offers opportunities to influence open-source communities and contribute to cutting-edge ML workloads.

• 5+ years of leading design or architecture of new and existing systems experience
• 2+ years of experience in developing compiler features and optimizations
• Proficiency with one or more programming languages: C++ (preferred), C, Python
• Master or PhD degree in computer science or equivalent
• Proficiency with resource management, scheduling, code generation, and compute graph optimization
• Experience optimizing TensorFlow, PyTorch, or JAX deep learning models
• Experience with multiple toolchains and Instruction Set Architectures

• Design, implement, test, deploy, and maintain innovative software solutions for the Neuron compiler.
• Solve hard compiler optimization problems to achieve optimum performance for various ML model families.
• Work with chip architects, runtime/OS engineers, scientists, and ML Apps 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, code reviews, and communicate with internal and external stakeholders.
• Build innovative features to enhance the user experience for developers globally.

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