Python Deep Learning Frameworks

Python Deep Learning Frameworks is a list of deep learning frameworks that are implemented in Python and hosted on GitHub. It is intended as an entry point for people new to deep learning in Python. The project is inspired by @mingrammer's Top Python Web Frameworks.

Analytics

Basic analytics of The List are available in this notebook.

The List

Project Name	Project Organization	Description	Stars	Forks	Open Issues	Last Commit
tensorflow	None	An Open Source Machine Learning Framework for Everyone	177663	88923	2041	0 day(s) ago
transformers	Hugging Face	🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.	111974	22237	776	0 day(s) ago
pytorch	None	Tensors and Dynamic neural networks in Python with strong GPU acceleration	70841	19481	12684	0 day(s) ago
keras	Keras	Deep Learning for humans	59344	19407	375	0 day(s) ago
caffe	Berkeley Vision and Learning Center	Caffe: a fast open framework for deep learning.	33553	18980	1181	0 day(s) ago
lightning	⚡️ Lightning AI	Deep learning framework to train, deploy, and ship AI products Lightning fast.	24637	3019	701	0 day(s) ago
Paddle	None	PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice （『飞桨』核心框架，深度学习&机器学习高性能单机、分布式训练和跨平台部署）	20847	5315	2135	0 day(s) ago
mxnet	The Apache Software Foundation	Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more	20558	6884	2005	0 day(s) ago
tinygrad	the tiny corp	You like pytorch? You like micrograd? You love tinygrad! ❤️	19426	2485	109	0 day(s) ago
albumentations	None	Fast image augmentation library and an easy-to-use wrapper around other libraries. Documentation: https://albumentations.ai/docs/ Paper about the library: https://www.mdpi.com/2078-2489/11/2/125	12547	1533	387	1 day(s) ago
tvm	The Apache Software Foundation	Open deep learning compiler stack for cpu, gpu and specialized accelerators	10317	3193	758	0 day(s) ago
sonnet	Google DeepMind	TensorFlow-based neural network library	9605	1360	36	1 day(s) ago
trax	Google	Trax — Deep Learning with Clear Code and Speed	7721	807	116	0 day(s) ago
SerpentAI	Serpent.AI	Game Agent Framework. Helping you create AIs / Bots that learn to play any game you own!	6566	785	2	0 day(s) ago
micrograd		A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API	6162	758	28	0 day(s) ago
chainer	Chainer	A flexible framework of neural networks for deep learning	5821	1402	12	1 day(s) ago
flax	Google	Flax is a neural network library for JAX that is designed for flexibility.	4689	538	179	0 day(s) ago
mlpack	mlpack	mlpack: a fast, header-only C++ machine learning library	4526	1522	24	1 day(s) ago
tensorforce	Tensorforce	Tensorforce: a TensorFlow library for applied reinforcement learning	3246	546	35	3 day(s) ago
catalyst	Catalyst-Team	Accelerated deep learning R&D	3168	398	4	6 day(s) ago
singa	The Apache Software Foundation	a distributed deep learning platform	2933	1047	51	5 day(s) ago
dm-haiku	Google DeepMind	JAX-based neural network library	2599	219	83	3 day(s) ago
lightly	Lightly	A python library for self-supervised learning on images.	2489	214	96	2 day(s) ago
coach	Intel Labs	Reinforcement Learning Coach by Intel AI Lab enables easy experimentation with state of the art Reinforcement Learning algorithms	2266	456	90	5 day(s) ago
karateclub		Karate Club: An API Oriented Open-source Python Framework for Unsupervised Learning on Graphs (CIKM 2020)	1972	231	7	5 day(s) ago
garage	Reinforcement Learning Working Group	A toolkit for reproducible reinforcement learning research.	1728	297	232	1 day(s) ago
detecto		Build fully-functioning computer vision models with PyTorch	600	109	45	0 day(s) ago
Neuraxle	Neuraxio	The world's cleanest AutoML library ✨ - Do hyperparameter tuning with the right pipeline abstractions to write clean deep learning production pipelines. Let your pipeline steps have hyperparameter spaces. Design steps in your pipeline like components. Compatible with Scikit-Learn, TensorFlow, and most other libraries, frameworks and MLOps environments.	580	60	30	0 day(s) ago
quickvision		An Easy To Use PyTorch Computer Vision Library	50	4	20	124 day(s) ago

Last update: Mon, 18 Sep 2023 08:22:50 UTC

Historical Data

As of 2021-09-30, this project also tracks the history of every repository in repositories.csv. Feel free to use it for your own data science needs. 😄

Contributing

If you want to add a new framework to the list, please add it to list.txt and open a PR!

shimst3r / python-deep-learning-frameworks Goto Github PK