分布式数学库开发工程师

职位描述

职责

1、基于C++开发计算引擎，主导单机多卡张量计算引擎的架构设计与核心开发。

2、实现DP/TP/MP/PP等并行模式，对Transformer/GNN等关键模块的系统级调优。

3、设计稠密数据的计算流方案，覆盖模型解析/图编译/IR转换/Pass优化等全链条流程开发和调优。

4、基于MLIR构建异构调度与内存管理体系，适配常用开源框架ONNX/pytorch。

任职要求

需求：

1、本科及以上学历，3年以上C++开发经验，具备扎实的计算机体系结构、操作系统与算法基础。

2、具备良好的工程素养，熟悉CMake构建与Git协作流程，能独立完成模块设计、开发与单元测试。

3、熟悉pytorch和MLIR，对graph/compile和pass等模块，有二次开发及优化经验。

4、深入理解深度学习分布式并行原理，熟练掌握DP/TP/MP/PP实现逻辑，有多卡分布式系统开发经验。