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.

21–40 of 213

Rank Project Stars Forks
21 Megatron-LM

Ongoing research training transformer models at scale

17.7K 4.4K
22 tvm

Open Machine Learning Compiler Framework

13.7K 4.0K
23 ggml

Tensor library for machine learning

15.3K 1.8K
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
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
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
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
28 stable-baselines3

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

13.7K 2.2K
29 litgpt

20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.

13.6K 1.5K
30 axolotl

Go ahead and axolotl questions

12.4K 1.4K
31 ludwig

Low-code framework for building custom LLMs, neural networks, and other AI models

11.7K 1.2K
32 segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

11.7K 1.8K
33 cutlass

CUDA Templates and Python DSLs for High-Performance Linear Algebra

10.3K 2.1K
34 Time-Series-Library

A Library for Advanced Deep Time Series Models for General Time Series Analysis.

12.8K 2.0K
35 machinelearning

ML.NET is an open source and cross-platform machine learning framework for .NET.

9.4K 1.9K
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
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
38 LAVIS

LAVIS - A One-stop Library for Language-Vision Intelligence

11.3K 1.1K
39 MONAI

AI Toolkit for Healthcare Imaging

8.6K 1.6K
40 einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

9.6K 398