Machine Learning - Compiler Engineer II, Annapurna Labs

AWS Machine Learning accelerators are leading AWS innovation in Generative AI, with the Inferentia chip delivering best-in-class ML inference performance at the lowest cloud cost. The role is within Amazon's Annapurna Labs team, responsible for AWS silicon development. As a Machine Learning Compiler Engineer II on the AWS Neuron team, you'll develop a compiler handling the world's largest ML workloads, working with frameworks like TensorFlow, PyTorch, and MXNET. The team values work-life balance and offers flexible working hours. You'll join a diverse, inclusive culture with strong mentorship opportunities and career growth potential. The position involves working on cutting-edge ML hardware acceleration, compiler optimization, and collaboration with top engineering talent. The team emphasizes knowledge sharing and professional development, with projects assigned to help members grow their skills. Amazon offers comprehensive benefits and a competitive compensation package, including equity and sign-on payments in addition to the base salary range.