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.

1–20 of 213

Rank Project Stars Forks
1 tensorflow

An Open Source Machine Learning Framework for Everyone

198.0K 76.2K
2 pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

102.7K 29.0K
3 keras

Deep Learning for humans

64.3K 19.8K
4 nanoGPT

The simplest, fastest repository for training/finetuning medium-sized GPTs.

62.6K 10.8K
5 ray

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

43.6K 8.0K
6 DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

43.0K 4.9K
7 ColossalAI

Making large AI models cheaper, faster and more accessible

41.4K 4.5K
8 spaCy

💫 Industrial-strength Natural Language Processing (NLP) in Python

33.9K 4.7K
9 pytorch-lightning

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

31.3K 3.8K
10 xgboost

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow

28.7K 8.9K
11 Paddle

PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice (『飞桨』核心框架,深度学习&机器学习高性能单机、分布式训练和跨平台部署)

24.1K 6.0K
12 fastai

The fastai deep learning library

28.1K 7.6K
13 pytorch_geometric

Graph Neural Network Library for PyTorch

24.0K 4.0K
14 llm.c

LLM training in simple, raw C/CUDA

30.9K 3.7K
15 caffe

Caffe: a fast open framework for deep learning.

34.6K 18.4K
16 onnx

Open standard for machine learning interoperability

21.4K 4.0K
17 peft

🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.

21.6K 2.5K
18 recommenders

Best Practices on Recommendation Systems

21.9K 3.3K
19 LightGBM

A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning tasks.

18.7K 4.1K
20 tfjs

A WebGL accelerated JavaScript library for training and deploying ML models.

19.1K 2.0K
< Previous
/ 11
Next >