分布式数学库开发工程师
北京 / 上海 / 合肥 / 成都 / 西安
社招 · 全职
工程开发
职位描述
职责
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实现逻辑，有多卡分布式系统开发经验。
