算子框架

面向科学智算“异构兼容、多层优化”需求,构建以共性算子为基元、计算库为支撑、求解模型为核心的三层框架体系。支持 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

国家标准已立项

《人工智能 科学计算 共性算子》

20252061-Z-469

查看国家标准 ↗

计算库

10 个专用计算库

共性算子

42 种算子 · 6 类 · 参考团标 T/CESA 1336-2024

国产硬件适配

算子在 8 款国产芯片上的验证状态

算子类别昇腾 NPU鲲鹏 CPU海光 CPU昆仑 XPU海光 DCU飞腾 CPU燧原 XPU沐曦 GPU
高阶微分操作24/2424/2424/2424/2424/2424/2424/2424/24
复数操作11/1111/1111/1111/1111/1111/1111/1111/11
积分操作3/33/33/33/33/33/33/33/3
傅里叶变换操作2/22/22/22/22/22/22/22/2
线性代数操作1/11/11/11/11/11/11/11/1
分数阶微分操作1/11/11/11/11/11/11/11/1
合计42/4242/4242/4242/4242/4242/4242/4242/42