Description
We are looking for a Deep Learning Compiler Engineer to join our team. As a member of our Deep Learning Compiler (DLC) team, you will be responsible for analyzing deep learning networks and developing compiler optimization algorithms. You will collaborate with members of the deep learning software framework teams and the hardware architecture teams to accelerate the next generation of deep learning software.
Your scope of work will include defining public APIs, performance optimizations and analysis, crafting and implementing compiler infrastructure techniques for neural networks, and other general software engineering work. You will also have the opportunity to work on projects that involve GPU kernel generation with high performance and fast build time.
To be successful in this role, you will need to have a strong background in computer science, computer engineering, or a related field, with a focus on performance analysis and compiler optimizations. You should have excellent C/C++ and Python programming and software design skills, including debugging, performance analysis, and test design.
If you are proficient in CPU and/or GPU architecture, CUDA or OpenCL programming experience, and have experience with technologies such as MLIR, XLA, TVM, LLVM, deep learning models and algorithms, and deep learning frameworks, such as PyTorch, you will be well-suited for this role.
As a senior engineer, you will be expected to lead your own development efforts, define project goals and scope, and work independently. You will also be expected to mentor junior engineers and interns, and contribute to the growth and development of the team.
We offer a highly competitive salary and a comprehensive benefits package, including equity and benefits. We are committed to creating a diverse and inclusive work environment, and we welcome applications from candidates of all backgrounds.
If you are interested in this opportunity, please submit your application by May 4, 2026. We look forward to hearing from you!