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.

141–160 of 213

Rank Project Stars Forks
141 Torch-Pruning

[CVPR 2023] DepGraph: Towards Any Structural Pruning; LLMs, Vision Foundation Models, etc.

3.3K 386
142 tab-transformer-pytorch

Implementation of TabTransformer, attention network for tabular data, in Pytorch

1.1K 131
143 Visual-RFT

Official repository of 'Visual-RFT: Visual Reinforcement Fine-Tuning' & 'Visual-ARFT: Visual Agentic Reinforcement Fine-Tuning'’

2.3K 111
144 TNN

A high-performance, lightweight cross-platform deep learning inference framework by Tencent, featuring model compression and code pruning.

4.6K 771
145 safe-rlhf

Safe RLHF: Constrained Value Alignment via Safe Reinforcement Learning from Human Feedback

1.6K 134
146 KuiperInfer

Implement a high-performance deep learning inference library step by step

3.5K 372
147 tflearn

Deep learning library featuring a higher-level API for TensorFlow.

9.6K 2.4K
148 d3rlpy

An offline deep reinforcement learning library

1.7K 267
149 Theano

Theano was a Python library that allows you to define, optimize, and evaluate mathematical expressions involving multi-dimensional arrays efficiently. It is being continued as PyTensor: www.github.com/pymc-devs/pytensor

10.0K 2.5K
150 LibMTL

A PyTorch Library for Multi-Task Learning

2.6K 232
151 EEG-DL

A Deep Learning library for EEG Tasks (Signals) Classification, based on TensorFlow.

1.2K 230
152 TensorFlow.NET

.NET Standard bindings for Google's TensorFlow for developing, training and deploying Machine Learning models in C# and F#.

3.4K 538
153 MegEngine

MegEngine is a fast, scalable, easy-to-use deep learning framework with automatic differentiation.

4.8K 545
154 gorgonia

Gorgonia is a library that helps facilitate machine learning in Go.

5.9K 450
155 MoBA

MoBA: Mixture of Block Attention for Long-Context LLMs

2.2K 159
156 ReCall

ReSearch: Learning to Reason with Search for LLMs via Reinforcement Learning & ReCall: Learning to Reason with Tool Call for LLMs via Reinforcement Learning

1.4K 89
157 GRPO-Zero

Implementing DeepSeek R1's GRPO algorithm from scratch

1.9K 99
158 spago

Self-contained Machine Learning and Natural Language Processing library in Go

1.9K 88
159 PRIME

Scalable RL solution for advanced reasoning of language models

1.9K 116
160 towhee

Towhee is a framework that is dedicated to making neural data processing pipelines simple and fast.

3.5K 259