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.

161–180 of 213

Rank Project Stars Forks
161 ktrain

ktrain is a Python library that makes deep learning and AI more accessible and easier to apply

1.3K 260
162 DataDreamer

DataDreamer: Prompt. Generate Synthetic Data. Train & Align Models.   🤖💤

1.1K 58
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
164 LongLoRA

Code and documents of LongLoRA and LongAlpaca (ICLR 2024 Oral)

2.7K 280
165 edward

A probabilistic programming language in TensorFlow. Deep generative models, variational inference.

4.8K 741
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
167 Transfer-Learning-Library

Transfer Learning Library for Domain Adaptation, Task Adaptation, and Domain Generalization

3.9K 602
168 dfdx

Deep learning in Rust, with shape checked tensors and neural networks

1.9K 108
169 DIG

A library for graph deep learning research

2.0K 289
170 OpenDelta

A plug-and-play library for parameter-efficient-tuning (Delta Tuning)

1.0K 83
171 chainer

A flexible framework of neural networks for deep learning

5.9K 1.3K
172 trlx

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

4.8K 486
173 mup

maximal update parametrization (µP)

1.8K 105
174 stellargraph

StellarGraph - Machine Learning on Graphs

3.1K 443
175 nanoT5

Fast & Simple repository for pre-training and fine-tuning T5-style models

1.0K 78
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
177 MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

3.0K 484
178 tensorpack

A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility

6.3K 1.8K
179 RL4LMs

A modular RL library to fine-tune language models to human preferences

2.4K 201
180 self-rewarding-lm-pytorch

Implementation of the training framework proposed in Self-Rewarding Language Model, from MetaAI

1.4K 70