深度学习框架

深度学习训练与推理核心框架,如 PyTorch、TensorFlow、JAX 等。

共 213 个项目

排名逻辑见 方法论 ;同类排名基于 GitHub 公开指标。

第 41–60 项,共 213 个

排名 项目 Stars Forks
41 pyro

基于Python和PyTorch的深度通用概率编程

9.0K 1.0K
42 sonnet

基于TensorFlow的神经网络库

10.0K 1.3K
43 dgl

用于在现有深度学习框架之上简化图深度学习的Python包。

14.3K 3.1K
44 bitsandbytes

利用PyTorch的k位量化,让大型语言模型更易用。

8.4K 918
45 slime

slime 是一个用于 RL 扩展的 LLM 后训练框架。

8.3K 1.2K
46 LMFlow

一个用于大型基础模型微调和推理的可扩展工具包。让大模型惠及所有人。

8.5K 825
47 deepchem

将深度学习普及到药物发现、量子化学、材料科学和生物学领域

7.0K 2.3K
48 autograd

高效计算NumPy代码的导数。

7.5K 940
49 Liger-Kernel

用于LLM训练的高效Triton内核

6.6K 588
50 gpt-neox

基于Megatron和DeepSpeed库,在GPU上实现模型并行自回归变换器。

7.5K 1.1K
51 smile

统计机器智能与学习引擎

6.4K 1.1K
52 angel

一个灵活而强大的参数服务器,用于大规模机器学习

6.8K 1.6K
53 skorch

一个与scikit-learn兼容的神经网络库,封装了PyTorch

6.2K 419
54 mergekit

用于合并预训练大型语言模型的工具。

7.3K 788
55 DALI

一个GPU加速的库,包含高度优化的构建模块和执行引擎,用于数据处理,以加速深度学习训练和推理应用。

5.7K 677
56 probability

TensorFlow中的概率推理和统计分析

4.4K 1.1K
57 AReaL

基于LLM的智能体应用的RL桥接。简单且灵活。

5.7K 589
58 x-transformers

一个简洁而完整的全注意力Transformer,集成了多篇论文中一系列有前景的实验性功能。

5.9K 517
59 ignite

用于在PyTorch中灵活透明地训练和评估神经网络的高级库。

4.8K 719
60 oneDNN

oneAPI深度神经网络库(oneDNN)

4.0K 1.2K
< 上一页
/ 11
下一页 >