About The Role

As a member of our Compiler team, you will work with leaders from industry and academia to develop entirely new solutions for the toughest problems in AI compute. As deep neural network architectures evolve, they are becoming enormously parallel, and distributed. Compilers are needed to optimize the mappings of computation graphs to compute nodes. In this position, you will build the tools that generate distributed memory code from evolving intermediate representations.

Responsibilities

• Design and devise graph semantics, intermediate representations, and abstraction layers between high-level definitions (like MLIR) and low-level (LLVM IR) distributed code
• Use state-of-the-art parallelization and partitioning techniques to automate generation of distributed kernels
• Low-level optimization on a SIMD/tensor-aware architecture of compute nodes
• Identify, design and implement novel program analysis and optimization techniques
• Design and implement custom system tools (such as linkers) for architectures with massive number of compute nodes

Requirements

• Enrolled in the University of Toronto's PEY program with a degree in Computer Science, Computer Engineering, or other related disciplines
• High proficiency in programming using Python or C++
• Solid understanding of fundamental concepts related to system design, such as data structures, algorithms, and operating systems.
• Related experience or fundamental knowledge of compilers and distributed systems
• Familiarity with high-level parallel program analysis and optimization

Preferred

• LLVM compiler internals
• Polyhedral models
• Familiarity with HPC kernels and their optimization