算子框架
面向科学智算“异构兼容、多层优化”需求,构建以共性算子为基元、计算库为支撑、求解模型为核心的三层框架体系。支持 PaddlePaddle、MindSpore、PyTorch 等主流框架异构混合适配,正建设 Jittor、OneFlow 等适配能力,兼容海光、昇腾等 8 款国产芯片,提供从高阶微分算子到领域求解模型的全栈能力。
科学任务
120 项 AI4Science 科学智算任务 · 12 个学科领域
↓
领域套件
按科学智算领域组织的专用工具集
↓
共性框架
算子库
42 种 · 6 类
计算库
10 类方法
模型库
预训练模型
科学智算框架
MindSpore华为昇腾
PaddlePaddle百度飞桨
PyTorch
↓
国产芯片
昇腾 NPU鲲鹏 CPU海光 CPU昆仑 XPU海光 DCU飞腾 CPU燧原 XPU沐曦 GPU
标准体系
团体标准
《人工智能 科学计算 共性算子》
T/CESA 1336-2024
计算库
10 个专用计算库
共性算子
42 种算子 · 6 类 · 参考团标 T/CESA 1336-2024
复数操作
11 种算子傅里叶变换操作
2 种算子线性代数操作
1 种算子分数阶微分操作
1 种算子国产硬件适配
算子在 8 款国产芯片上的验证状态
| 算子类别 | 昇腾 NPU | 鲲鹏 CPU | 海光 CPU | 昆仑 XPU | 海光 DCU | 飞腾 CPU | 燧原 XPU | 沐曦 GPU |
|---|---|---|---|---|---|---|---|---|
| 高阶微分操作 | 24/24 | 24/24 | 24/24 | 24/24 | 24/24 | 24/24 | 24/24 | 24/24 |
| 复数操作 | 11/11 | 11/11 | 11/11 | 11/11 | 11/11 | 11/11 | 11/11 | 11/11 |
| 积分操作 | 3/3 | 3/3 | 3/3 | 3/3 | 3/3 | 3/3 | 3/3 | 3/3 |
| 傅里叶变换操作 | 2/2 | 2/2 | 2/2 | 2/2 | 2/2 | 2/2 | 2/2 | 2/2 |
| 线性代数操作 | 1/1 | 1/1 | 1/1 | 1/1 | 1/1 | 1/1 | 1/1 | 1/1 |
| 分数阶微分操作 | 1/1 | 1/1 | 1/1 | 1/1 | 1/1 | 1/1 | 1/1 | 1/1 |
| 合计 | 42/42 | 42/42 | 42/42 | 42/42 | 42/42 | 42/42 | 42/42 | 42/42 |
