Deep Learning Frameworks

Core deep learning training and inference frameworks — PyTorch, TensorFlow, JAX, and similar stacks.

213 projects

See methodology for ranking rules; order uses public GitHub metrics within this scenario.

41–60 of 213

Rank Project Stars Forks
41 pyro

Deep universal probabilistic programming with Python and PyTorch

9.0K 1.0K
42 sonnet

TensorFlow-based neural network library

10.0K 1.3K
43 dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.

14.3K 3.1K
44 bitsandbytes

Accessible large language models via k-bit quantization for PyTorch.

8.4K 918
45 slime

slime is an LLM post-training framework for RL Scaling.

8.3K 1.2K
46 LMFlow

An Extensible Toolkit for Finetuning and Inference of Large Foundation Models. Large Models for All.

8.5K 825
47 deepchem

Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, Materials Science and Biology

7.0K 2.3K
48 autograd

Efficiently computes derivatives of NumPy code.

7.5K 940
49 Liger-Kernel

Efficient Triton Kernels for LLM Training

6.6K 588
50 gpt-neox

An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries

7.5K 1.1K
51 smile

Statistical Machine Intelligence & Learning Engine

6.4K 1.1K
52 angel

A Flexible and Powerful Parameter Server for large-scale machine learning

6.8K 1.6K
53 skorch

A scikit-learn compatible neural network library that wraps PyTorch

6.2K 419
54 mergekit

Tools for merging pretrained large language models.

7.3K 788
55 DALI

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

5.7K 677
56 probability

Probabilistic reasoning and statistical analysis in TensorFlow

4.4K 1.1K
57 AReaL

The RL Bridge for LLM-based Agent Applications. Made Simple & Flexible.

5.7K 589
58 x-transformers

A concise but complete full-attention transformer with a set of promising experimental features from various papers

5.9K 517
59 ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

4.8K 719
60 oneDNN

oneAPI Deep Neural Network Library (oneDNN)

4.0K 1.2K