AI Compiler

The AI Compiler team is central to developing the means by which cutting-edge AI models and workflows achieve world-record results on our unique hardware architecture.

Requirements

• 4+ years of experience in software engineering.
• 2+ years of experience developing a backend for a large-scale, open-source-based compiler (e.g. LLVM)
• At least one of the following:
• Demonstrable open-source contribution to relevant projects.
• Experience developing state-of-the-art neural network architectures.
• Specialization experience in ML accelerators, high performance computing, GPU and/or CPU architectures.
• Backend experience with MLIR: an advantage.

Responsibilities

• Develop a state-of-the-art MLIR-based AI compiler stack targeting our next generation hardware and delivering superior acceleration to AI applications.
• Interface with platform teams to define IRs leveraging unique our hardware architecture.
• Collaborate with runtime teams to shape the future of our runtime stack, from driver level up to integration into popular AI frameworks.
• Collaborate with toolchain teams to deliver a holistic toolchain for our customers, supporting static and dynamic development workflows.