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 knowledge-sharing and mentorship, providing a supportive environment for career growth. You will work closely with chip architects and ML application teams to enhance compiler performance and user experience. The position involves solving complex optimization problems and collaborating with open-source communities.

• Bachelor's degree in computer science or equivalent.
• 5+ years of full software development life cycle experience.
• 3+ years of experience in developing compiler features and optimizations.
• Proficiency in C++ or Python.
• Experience with LLVM and/or MLIR.
• Experience in LLM, Vision, or other deep-learning models.

• Design, implement, test, deploy, and maintain innovative software solutions for the Neuron compiler.
• Collaborate with chip architects, runtime engineers, and ML application teams to deploy ML models on AWS accelerators.
• Create compiler optimization and verification passes to enhance performance and stability.
• Participate in design discussions and code reviews with internal and external stakeholders.
• Work in a startup-like environment focusing on critical tasks and projects.

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