Amazon - Auburn, AL

posted about 2 months ago

Full-time - Mid Level
Auburn, AL
Sporting Goods, Hobby, Musical Instrument, Book, and Miscellaneous Retailers

About the position

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 understand large language model (LLM) training processes, which are integral to the optimization of application models. This position requires a proactive mindset and a commitment to innovation in the field of deep learning compilers.

Responsibilities

  • Develop and optimize a deep learning compiler stack for various application models.
  • Build fuzzers and specification synthesis tooling for the LLVM-based compiler.
  • Collaborate with a team focused on scientific approaches to software engineering problems.
  • Push the boundaries of compiler technology to deliver optimal solutions for customers.
  • Apply strong software development skills in C++ and Python to compiler or verification projects.

Requirements

  • Proficiency in C++ and Python programming, particularly in compiler or verification projects.
  • Demonstrated scientific approach to software engineering problems.
  • Master's degree or PhD in computer science or equivalent.
  • Understanding of large language model (LLM) training processes.

Benefits

  • Equity compensation
  • Sign-on payments
  • Full range of medical benefits
  • Financial benefits
  • Other forms of compensation
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service