The AWS Neuron Compiler team is looking for skilled compiler engineers to develop a cutting-edge deep learning compiler stack. This role involves optimizing application models from frameworks like PyTorch, TensorFlow, and JAX, while working with AWS's custom Machine Learning accelerators. As a ML Compiler engineer, you will design, develop, and optimize features for the compiler, collaborating with cross-functional teams to enhance system-wide performance. The position emphasizes mentorship and career growth within a supportive team environment.

• Bachelor's degree in Computer Science, Engineering, Mathematics, or a related field
• 3+ years of non-internship professional software development experience
• Proficiency with 1 or more programming languages, preferably C++
• 2+ years of experience in optimization algorithms, graph-theory, or hardware resource management
• 3+ years of full software development life cycle experience, including coding standards and testing
• M.S. or Ph.D. in computer science or related field
• Strong knowledge in compiler design, instruction scheduling, and memory allocation
• Experience with LLVM and/or MLIR
• Experience with developing algorithms for simulation tools
• Experience with TensorFlow, PyTorch, and/or JAX
• Experience in LLM, Vision, or other deep-learning models

• Solve challenging technical problems at every layer of the stack.
• Design, implement, test, deploy, and maintain innovative software solutions.
• Research implementations that deliver the best possible experiences for customers.
• Build high-impact solutions for a large customer base.
• Participate in design discussions and code reviews.
• Work cross-functionally to drive business decisions with technical input.
• Create metrics and implement automation to improve software architecture.
• Resolve the root cause of software defects.

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