Sr. Deep Learning Compiler Engineer III, AWS Neuron, Annapurna Labs

AWS Neuron team is seeking a Senior Deep Learning Compiler Engineer to join Annapurna Labs, focusing on revolutionizing deep learning accessibility. The role involves working with AWS Neuron SDK, which optimizes neural net models' performance on AWS Inferentia and Trainium custom chips. You'll be part of a team developing cutting-edge compiler technology for machine learning workloads, working at the intersection of ML, high-performance computing, and distributed architectures. The position offers unique opportunities to influence AWS's machine learning infrastructure, collaborate with diverse teams, and solve complex technical challenges. The team operates in a startup-like environment while backed by AWS's robust infrastructure and resources. You'll contribute to pre-silicon design, product development, and innovation in ML acceleration. AWS provides an inclusive culture, emphasizing work-life balance, mentorship, and professional growth. The role combines technical leadership with hands-on development, requiring expertise in compiler development and an understanding of machine learning systems. You'll work with AWS ML services teams, contribute to product strategy, and help shape the future of machine learning infrastructure at AWS.