深度学习框架

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

共 213 个项目

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

第 61–80 项,共 213 个

排名 项目 Stars Forks
61 SynapseML

简单分布式机器学习Python库,为Spark移植ML算法

5.2K 868
62 tch-rs

PyTorch的C++ API的Rust绑定。

5.5K 452
63 oneflow

OneFlow是一个深度学习框架,设计上注重用户友好、可扩展和高效率。

9.4K 1.0K
64 nccl

多GPU集体通信的优化原语

5.0K 1.4K
65 Flux.jl

放松!Flux 是一个不会让你紧张的机器学习库。

4.7K 625
66 djl

一个引擎无关的Java深度学习框架

4.8K 760
67 xtuner

为超大规模MoE模型打造的下一代训练引擎

5.2K 447
68 autokeras

用于深度学习的AutoML库

9.3K 1.4K
69 mmf

来自Facebook AI Research (FAIR)的视觉与语言多模态研究的模块化框架。

5.6K 938
70 EasyR1

EasyR1:基于veRL的高效、可扩展的多模态强化学习训练框架

5.1K 390
71 kaolin

一个加速3D深度学习研究的PyTorch库

5.2K 628
72 flashlight

用于机器学习的C++独立库

5.5K 501
73 deepxde

用于科学机器学习和物理信息学习的库

4.4K 990
74 rl

一个模块化、原语优先、Python优先的强化学习PyTorch库。

3.5K 478
75 TransformerEngine

一个在NVIDIA GPU上加速Transformer模型的库,包括在Hopper、Ada和Blackwell GPU上使用8位和4位浮点(FP8和FP4)精度,在训练和推理中提供更好的性能并降低内存占用。

3.5K 814
76 composer

提升你的模型训练

5.5K 467
77 PettingZoo

多智能体强化学习环境的标准API,包含流行的参考环境和相关工具

3.5K 520
78 vector-quantize-pytorch

向量(和标量)量化,Pytorch实现

4.0K 337
79 dm-haiku

基于JAX的神经网络库

3.3K 299
80 physicsnemo

开源深度学习框架,用于使用最先进的Physics-ML方法构建、训练和微调深度学习模型。

3.2K 766
< 上一页
/ 11
下一页 >