Senior ML Compiler Engineer, AWS Neuron Team

Become a Senior ML Compiler Engineer at Amazon Web Services (AWS) and help evolve the AWS Neuron SDK for deep learning applications. Focus on developing and optimizing high-performance kernels for machine learning.

In the Annapurna Labs team, you will contribute significantly to the Acceleration Kernel Library, tasked with driving AI hardware and software enhancements. This role features hands-on kernel design, extensive performance analysis, and customer collaboration. Work in a flexible and inclusive environment that values mentorship and professional growth.

Key Responsibilities:

• Design and implement ML computation kernels leveraging Neuron architecture
• Execute thorough performance optimization across multiple hardware generations
• Use profiling tools for performance bottleneck analysis
• Implement compiler optimizations such as fusion and tiling
• Collaborate with customers for optimizing their models

Requirements

• 3+ years of software development experience
• 2+ years in architecture and design roles
• Proficient in at least one programming language
• Strong understanding of coding standards and practices
• Bachelor's degree in computer science or equivalent preferred

Elevate your career by shaping the future of AI with AWS, developing cutting-edge solutions for a global audience.