工作職責:
1、基于自研 AI 芯片編譯器的架構設計及開發(fā);
2、 AI 編譯器各類 Pass 設計與開發(fā);
3、模型性能調(diào)優(yōu)工具的設計及開發(fā);
4、算子開發(fā)與優(yōu)化。
任職要求:
1、本科及以上專業(yè),良好的 C/C++ 編程功底,扎實的數(shù)據(jù)結構與算法基礎;
2、熟悉NPU/GPU 等硬件計算架構;
3、熟悉深度學習編譯,有 MLIR/IREE/TVM 等開發(fā)經(jīng)驗;
4、良好的團隊溝通能力與問題解決能力;
5、任意滿足如下條件之一者優(yōu)先:
a)有 MLIR/TVM/LLVM/XLA/IREE 等編譯器開發(fā)經(jīng)驗;
b)熟悉 CUDA Triton Cutlass 等 GPU 編程或者其他 NPU 編程界面;
c)熟悉計算機體系結構,熟悉芯片架構、指令集等;
d) NOI/ICPC/CCPC 競賽獲獎。