深度学习框架

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

共 213 个项目

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

第 1–20 项,共 213 个

排名 项目 Stars Forks
1 tensorflow

一个面向所有人的开源机器学习框架

198.0K 76.2K
2 pytorch

Python中的张量和动态神经网络,具有强大的GPU加速

102.7K 29.0K
3 keras

深度学习,以人为本

64.3K 19.8K
4 nanoGPT

用于训练/微调中型GPT的最简单、最快的仓库。

62.6K 10.8K
5 ray

Ray是一个AI计算引擎。Ray包含一个核心分布式运行时和一组用于加速机器学习工作负载的AI库。

43.6K 8.0K
6 DeepSpeed

DeepSpeed是一个深度学习优化库,使分布式训练和推理变得简单、高效和有效。

43.0K 4.9K
7 ColossalAI

让大型AI模型更便宜、更快、更易获取

41.4K 4.5K
8 spaCy

💫 工业级自然语言处理(NLP)Python库

33.9K 4.7K
9 pytorch-lightning

在1个或10,000+个GPU上预训练、微调任意大小的AI模型,零代码更改。

31.3K 3.8K
10 xgboost

可扩展、可移植、分布式的梯度提升(GBDT、GBRT或GBM)库,支持Python、R、Java、Scala、C等语言。可运行于单机、Hadoop、Spark、Dask、Flink和DataFlow。

28.7K 8.9K
11 Paddle

并行分布式深度学习:来自工业实践的机器学习框架(『飞桨』核心框架,深度学习&机器学习高性能单机、分布式训练和跨平台部署)

24.1K 6.0K
12 fastai

fastai深度学习库

28.1K 7.6K
13 pytorch_geometric

PyTorch的图神经网络库

24.0K 4.0K
14 llm.c

用简单、原始的C/CUDA训练LLM

30.9K 3.7K
15 caffe

Caffe:一个快速的深度学习开源框架。

34.6K 18.4K
16 onnx

机器学习互操作性的开放标准

21.4K 4.0K
17 peft

🤗 PEFT:最先进的参数高效微调技术。

21.6K 2.5K
18 recommenders

推荐系统最佳实践

21.9K 3.3K
19 LightGBM

一个基于决策树算法的快速、分布式、高性能梯度提升(GBT、GBDT、GBRT、GBM或MART)框架,用于排序、分类及许多其他机器学习任务。

18.7K 4.1K
20 tfjs

一个用于训练和部署机器学习模型的WebGL加速JavaScript库。

19.1K 2.0K
< 上一页
/ 11
下一页 >