Amazon - Mercer Island, 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. Your work will be crucial in pushing the boundaries of what is known in the field, ensuring that we deliver the best possible solutions to our customers. In this role, you will collaborate with a team that emphasizes a scientific approach to software engineering. Strong software development skills using C++ and Python are critical to this position. A background in compiler development is strongly preferred, as you will be applying your expertise to enhance our compiler's capabilities. You will also be expected to demonstrate a scientific approach to solving complex software engineering problems, contributing to the overall success of the team and the projects we undertake. Candidates must be currently located in 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 proficiency in C++ and Python programming, particularly in the context of compiler or verification projects. Additionally, an understanding of large language model (LLM) training processes will be beneficial in this role.