01 / RESPONSIBILITIES
岗位职责
- 将 ONNX/PyTorch 等模型图转换为设备可执行的 subgraph 和 Artifact;
- 完成RISC-V 处理器、光学矩阵计算单元与主机之间的图切分、算子融合和回退执行;
- 建立 NumPy/PyTorch golden model、算子测试和端到端数值回归;
- 与 Runtime、Firmware和 FPGA 原型工程师共同打通模型到硬件结果的执行链路。
02 / QUALIFICATIONS
任职要求
- 熟悉 C++,能够使用 Python 完成模型处理、测试或自动化工作;
- 熟悉 LLVM、MLIR、TVM、XLA 或其他编译器基础设施之一;
- 理解 IR、pass、pattern rewrite、graph partition、fusion 和 code generation;
- 具备 AI 模型编译、算子优化或高性能 kernel 开发经验(加分项)。
