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.
| Rank | Project | Stars | Forks | Updated | License |
|---|---|---|---|---|---|
| 161 |
ktrain
ktrain is a Python library that makes deep learning and AI more accessible and easier to apply |
1.3K | 260 | 02/06/25 | Apache-2.0 |
| 162 |
DataDreamer
DataDreamer: Prompt. Generate Synthetic Data. Train & Align Models. 🤖💤 |
1.1K | 58 | 02/02/25 | MIT |
| 163 |
higgsfield
Fault-tolerant, highly scalable GPU orchestration, and a machine learning framework designed for training models with billions to trillions of parameters |
4.1K | 708 | 05/25/24 | Apache-2.0 |
| 164 |
LongLoRA
Code and documents of LongLoRA and LongAlpaca (ICLR 2024 Oral) |
2.7K | 280 | 08/14/24 | Apache-2.0 |
| 165 |
edward
A probabilistic programming language in TensorFlow. Deep generative models, variational inference. |
4.8K | 741 | 03/18/24 | Other |
| 166 |
EasyLM
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax. |
2.5K | 259 | 08/13/24 | Apache-2.0 |
| 167 |
Transfer-Learning-Library
Transfer Learning Library for Domain Adaptation, Task Adaptation, and Domain Generalization |
3.9K | 602 | 05/10/24 | MIT |
| 168 |
dfdx
Deep learning in Rust, with shape checked tensors and neural networks |
1.9K | 108 | 07/23/24 | Other |
| 169 |
DIG
A library for graph deep learning research |
2.0K | 289 | 07/15/24 | GPL-3.0 |
| 170 |
OpenDelta
A plug-and-play library for parameter-efficient-tuning (Delta Tuning) |
1.0K | 83 | 09/19/24 | Apache-2.0 |
| 171 |
chainer
A flexible framework of neural networks for deep learning |
5.9K | 1.3K | 08/28/23 | MIT |
| 172 |
trlx
A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF) |
4.8K | 486 | 01/08/24 | MIT |
| 173 |
mup
maximal update parametrization (µP) |
1.8K | 105 | 07/17/24 | MIT |
| 174 |
stellargraph
StellarGraph - Machine Learning on Graphs |
3.1K | 443 | 04/10/24 | Apache-2.0 |
| 175 |
nanoT5
Fast & Simple repository for pre-training and fine-tuning T5-style models |
1.0K | 78 | 08/21/24 | Apache-2.0 |
| 176 |
graph4nlp
Graph4nlp is the library for the easy use of Graph Neural Networks for NLP. Welcome to visit our DLG4NLP website (https://dlg4nlp.github.io/index.html) for various learning resources! |
1.7K | 206 | 06/24/24 | Apache-2.0 |
| 177 |
MinkowskiEngine
Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors |
3.0K | 484 | 03/05/24 | Other |
| 178 |
tensorpack
A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility |
6.3K | 1.8K | 08/06/23 | Apache-2.0 |
| 179 |
RL4LMs
A modular RL library to fine-tune language models to human preferences |
2.4K | 201 | 03/01/24 | Apache-2.0 |
| 180 |
self-rewarding-lm-pytorch
Implementation of the training framework proposed in Self-Rewarding Language Model, from MetaAI |
1.4K | 70 | 04/11/24 | MIT |