深度学习框架
深度学习训练与推理核心框架,如 PyTorch、TensorFlow、JAX 等。
共 213 个项目
排名逻辑见 方法论 ;同类排名基于 GitHub 公开指标。
| 排名 | 项目 | Stars | Forks | 更新 | 许可证 |
|---|---|---|---|---|---|
| 41 |
pyro
基于Python和PyTorch的深度通用概率编程 |
9.0K | 1.0K | 08/27/26 | Apache-2.0 |
| 42 |
sonnet
基于TensorFlow的神经网络库 |
10.0K | 1.3K | 07/07/26 | Apache-2.0 |
| 43 |
dgl
用于在现有深度学习框架之上简化图深度学习的Python包。 |
14.3K | 3.1K | 07/31/25 | Apache-2.0 |
| 44 |
bitsandbytes
利用PyTorch的k位量化,让大型语言模型更易用。 |
8.4K | 918 | 08/27/26 | MIT |
| 45 |
slime
slime 是一个用于 RL 扩展的 LLM 后训练框架。 |
8.3K | 1.2K | 08/28/26 | Apache-2.0 |
| 46 |
LMFlow
一个用于大型基础模型微调和推理的可扩展工具包。让大模型惠及所有人。 |
8.5K | 825 | 08/10/26 | Apache-2.0 |
| 47 |
deepchem
将深度学习普及到药物发现、量子化学、材料科学和生物学领域 |
7.0K | 2.3K | 08/20/26 | MIT |
| 48 |
autograd
高效计算NumPy代码的导数。 |
7.5K | 940 | 08/29/26 | MIT |
| 49 |
Liger-Kernel
用于LLM训练的高效Triton内核 |
6.6K | 588 | 08/28/26 | BSD-2-Clause |
| 50 |
gpt-neox
基于Megatron和DeepSpeed库,在GPU上实现模型并行自回归变换器。 |
7.5K | 1.1K | 06/11/26 | Apache-2.0 |
| 51 |
smile
统计机器智能与学习引擎 |
6.4K | 1.1K | 08/28/26 | Other |
| 52 |
angel
一个灵活而强大的参数服务器,用于大规模机器学习 |
6.8K | 1.6K | 07/26/26 | Other |
| 53 |
skorch
一个与scikit-learn兼容的神经网络库,封装了PyTorch |
6.2K | 419 | 08/27/26 | BSD-3-Clause |
| 54 |
mergekit
用于合并预训练大型语言模型的工具。 |
7.3K | 788 | 06/17/26 | LGPL-3.0 |
| 55 |
DALI
一个GPU加速的库,包含高度优化的构建模块和执行引擎,用于数据处理,以加速深度学习训练和推理应用。 |
5.7K | 677 | 08/28/26 | Apache-2.0 |
| 56 |
probability
TensorFlow中的概率推理和统计分析 |
4.4K | 1.1K | 08/06/26 | Apache-2.0 |
| 57 |
AReaL
基于LLM的智能体应用的RL桥接。简单且灵活。 |
5.7K | 589 | 08/29/26 | Apache-2.0 |
| 58 |
x-transformers
一个简洁而完整的全注意力Transformer,集成了多篇论文中一系列有前景的实验性功能。 |
5.9K | 517 | 08/28/26 | MIT |
| 59 |
ignite
用于在PyTorch中灵活透明地训练和评估神经网络的高级库。 |
4.8K | 719 | 08/28/26 | BSD-3-Clause |
| 60 |
oneDNN
oneAPI深度神经网络库(oneDNN) |
4.0K | 1.2K | 08/29/26 | Apache-2.0 |