4. AI For Science 发布列表
名称 |
类型 |
对应官方版本 |
支持芯片系列 |
开发者社区发布情况 |
Python 版本 |
训练/推理框架版本 |
|---|---|---|---|---|---|---|
PhysicsNeMo |
AI for Science 框架 |
1.0.1 |
C500 |
镜像，二进制包 |
3.10 |
Pytorch 2.4 |
DeepXDE |
AI for Science 框架 |
1.14.0 |
C500 |
镜像 |
3.10 |
Pytorch 2.4/Paddle 2.6 |
Warp |
科学计算 Python 框架 |
1.8.1 |
C500 |
镜像，二进制包 |
3.10 |
Pytorch 2.4 |
DGL |
图神经网络框架 |
2.4 |
C500 |
镜像 |
3.10 |
Pytorch 2.4 |
PyG |
图神经网络框架 |
2.7.0 |
C500 |
镜像，二进制包 |
3.10 |
Pytorch 2.4 |
Boltz-2 |
结合自由能预测模型 |
2.2.0 |
C500 |
镜像 |
3.10 |
Pytorch 2.4 |
DiffDock |
分子对接模型 |
1.1.3 |
C500 |
镜像 |
3.10 |
Pytorch 2.4 |
ai-models |
气象模型集合，包含 AIFS、Aurora、FengWu、FourCastNet、FuXi、GenCast、GraphCast 和 Pangu-Weather气象模型 |
0.7.4 |
C500 |
镜像 |
3.12 |
Pytorch 2.4 |
NeuralGCM |
气象模型 |
1.2.2 |
C500 |
镜像 |
3.12 |
Jax 3.5.3.0 |