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 |
|---|---|---|---|---|---|
| 21 |
Megatron-LM
Ongoing research training transformer models at scale |
17.7K | 4.4K | 08/29/26 | Other |
| 22 |
tvm
Open Machine Learning Compiler Framework |
13.7K | 4.0K | 08/29/26 | Apache-2.0 |
| 23 |
ggml
Tensor library for machine learning |
15.3K | 1.8K | 08/25/26 | MIT |
| 24 |
burn
Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability. |
15.8K | 1.0K | 08/29/26 | Apache-2.0 |
| 25 |
transformers.js
State-of-the-art Machine Learning for the web. Run 🤗 Transformers directly in your browser, with no need for a server! |
16.3K | 1.2K | 08/29/26 | Apache-2.0 |
| 26 |
ms-swift
Use PEFT or Full-parameter to CPT/SFT/DPO/GRPO 600+ LLMs (Qwen3.6, DeepSeek-V4, GLM-5.1, InternLM3, Llama4, ...) and 300+ MLLMs (Qwen3-VL, Qwen3-Omni, InternVL3.5, Ovis2.5, GLM4.5v, Gemma4, Llava, Phi4, ...) (AAAI 2025). |
15.4K | 1.6K | 08/29/26 | Apache-2.0 |
| 27 |
deeplearning4j
JVM-based deep learning toolkit with model import for Keras, TensorFlow, and ONNX/PyTorch, plus ND4J linear algebra and SameDiff autodiff. |
14.2K | 3.8K | 08/28/26 | Apache-2.0 |
| 28 |
stable-baselines3
PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms. |
13.7K | 2.2K | 08/17/26 | MIT |
| 29 |
litgpt
20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale. |
13.6K | 1.5K | 08/17/26 | Apache-2.0 |
| 30 |
axolotl
Go ahead and axolotl questions |
12.4K | 1.4K | 08/29/26 | Apache-2.0 |
| 31 |
ludwig
Low-code framework for building custom LLMs, neural networks, and other AI models |
11.7K | 1.2K | 08/24/26 | Apache-2.0 |
| 32 |
segmentation_models.pytorch
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones. |
11.7K | 1.8K | 08/24/26 | MIT |
| 33 |
cutlass
CUDA Templates and Python DSLs for High-Performance Linear Algebra |
10.3K | 2.1K | 08/28/26 | Other |
| 34 |
Time-Series-Library
A Library for Advanced Deep Time Series Models for General Time Series Analysis. |
12.8K | 2.0K | 04/18/26 | MIT |
| 35 |
machinelearning
ML.NET is an open source and cross-platform machine learning framework for .NET. |
9.4K | 1.9K | 08/28/26 | MIT |
| 36 |
OpenRLHF
An Easy-to-use, Scalable and High-performance Agentic RL Framework based on Ray (PPO & DAPO & REINFORCE++ & VLM & TIS & vLLM & Ray & Async RL) |
10.0K | 1.0K | 08/13/26 | Apache-2.0 |
| 37 |
vowpal_wabbit
Vowpal Wabbit is a machine learning system which pushes the frontier of machine learning with techniques such as online, hashing, allreduce, reductions, learning2search, active, and interactive learning. |
8.7K | 1.9K | 08/26/26 | Other |
| 38 |
LAVIS
LAVIS - A One-stop Library for Language-Vision Intelligence |
11.3K | 1.1K | 06/02/26 | BSD-3-Clause |
| 39 |
MONAI
AI Toolkit for Healthcare Imaging |
8.6K | 1.6K | 08/29/26 | Apache-2.0 |
| 40 |
einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others) |
9.6K | 398 | 08/26/26 | MIT |