Machine Learning - Compiler Engineer II, Annapurna Labs

The AWS Neuron team is developing a deep learning compiler stack that takes neural network descriptions created in frameworks such as TensorFlow, PyTorch, and MXNET, and converts them into code suitable for execution. As a Machine Learning Compiler Engineer II on the AWS Neuron team, you will be supporting the ground-up development and scaling of a compiler to handle the world's largest ML workloads. You will be architecting and implementing business-critical features, publishing cutting-edge research, and contributing to a brilliant team of experienced engineers. You will leverage your technical communications skill as a hands-on partner to AWS ML services teams and will be involved in pre-silicon design, bringing new products/features to market, and many other exciting projects. The team values work-life balance and offers flexibility in working hours. They are dedicated to supporting new members, with a broad mix of experience levels and tenures, and are building an environment that celebrates knowledge sharing and mentorship.