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