Amazon - Maple Valley, WA
posted about 2 months ago
The AWS Neuron Compiler team is actively seeking skilled compiler engineers to join our efforts in developing a state-of-the-art deep learning compiler stack. This stack is designed to optimize application models across diverse domains, including Large Language and Vision, originating from leading frameworks such as PyTorch, TensorFlow, and JAX. As a member of this team, you will help lead the efforts in building fuzzers and specification synthesis tooling for our LLVM-based compiler. This role requires a strong focus on scientific approaches to software engineering problems, pushing the boundaries of what is known to deliver the best results for our customers. In this position, you will work collaboratively within a team that emphasizes scientific rigor and innovation. Your contributions will be critical in enhancing the performance and capabilities of the compiler stack, ensuring it meets the needs of various applications in the deep learning space. Strong software development skills using C++ and Python are essential, as you will be applying these languages to compiler or verification projects. A background in compiler development is strongly preferred, as it will enable you to navigate the complexities of the compiler stack effectively. Candidates must be currently located or willing to relocate to Seattle (preferred), Cupertino, Austin, or Toronto. A Master's degree or PhD in computer science or a related field is required, along with a solid understanding of large language model (LLM) training processes. This position offers a unique opportunity to work at the forefront of deep learning technology, contributing to advancements that will shape the future of AI applications.