Best-of Machine Learning with Python

🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.

This curated list contains 920 awesome open-source projects with a total of 4.8M stars grouped into 34 categories. All projects are ranked by a project-quality score, which is calculated based on various metrics automatically collected from GitHub and different package managers. If you like to add or update projects, feel free to open an issue, submit a pull request, or directly edit the projects.yaml. Contributions are very welcome!

🧙‍♂️ Discover other best-of lists or create your own.
📫 Subscribe to our newsletter for updates and trending projects.

Machine Learning Frameworks 63 projects
Data Visualization 55 projects
Text Data & NLP 103 projects
Image Data 64 projects
Graph Data 36 projects
Audio Data 29 projects
Geospatial Data 22 projects
Financial Data 25 projects
Time Series Data 29 projects
Medical Data 19 projects
Tabular Data 5 projects
Optical Character Recognition 12 projects
Data Containers & Structures 1 projects
Data Loading & Extraction 1 projects
Web Scraping & Crawling 1 projects
Data Pipelines & Streaming 2 projects
Distributed Machine Learning 36 projects
Hyperparameter Optimization & AutoML 52 projects
Reinforcement Learning 23 projects
Recommender Systems 17 projects
Privacy Machine Learning 7 projects
Workflow & Experiment Tracking 40 projects
Model Serialization & Deployment 20 projects
Model Interpretability 54 projects
Vector Similarity Search (ANN) 13 projects
Probabilistics & Statistics 24 projects
Adversarial Robustness 9 projects
GPU & Accelerator Utilities 20 projects
Tensorflow Utilities 16 projects
Jax Utilities 3 projects
Sklearn Utilities 19 projects
Pytorch Utilities 32 projects
Database Clients 1 projects
Others 66 projects

Explanation

🥇🥈🥉 Combined project-quality score
⭐️ Star count from GitHub
🐣 New project (less than 6 months old)
💤 Inactive project (6 months no activity)
💀 Dead project (12 months no activity)
📈📉 Project is trending up or down
➕ Project was recently added
❗️ Warning (e.g. missing/risky license)
👨‍💻 Contributors count from GitHub
🔀 Fork count from GitHub
📋 Issue count from GitHub
⏱️ Last update timestamp on package manager
📥 Download count from package manager
📦 Number of dependent projects
Tensorflow related project
Sklearn related project
PyTorch related project
MxNet related project
Apache Spark related project
Jupyter related project
PaddlePaddle related project
Pandas related project
Jax related project

Machine Learning Frameworks

General-purpose machine learning and deep learning frameworks.

Tensorflow (🥇55 · ⭐ 190K) - An Open Source Machine Learning Framework for Everyone. Apache-2

GitHub (👨‍💻 4.8K · 🔀 74K · 📦 460K · 📋 46K - 15% open · ⏱️ 13.02.2025):
```
	git clone https://github.com/tensorflow/tensorflow
```
PyPi (📥 20M / month · 📦 8.4K · ⏱️ 26.01.2025):
```
	pip install tensorflow
```

Conda (📥 5.3M · ⏱️ 10.02.2025):

	conda install -c conda-forge tensorflow

Docker Hub (📥 79M · ⭐ 2.7K · ⏱️ 13.02.2025):
```
	docker pull tensorflow/tensorflow
```

PyTorch (🥇55 · ⭐ 87K) - Tensors and Dynamic neural networks in Python with strong GPU.. BSD-3

GitHub (👨‍💻 5.4K · 🔀 23K · 📥 75K · 📦 640K · 📋 50K - 31% open · ⏱️ 13.02.2025):

	git clone https://github.com/pytorch/pytorch

PyPi (📥 39M / month · 📦 22K · ⏱️ 29.01.2025):

	pip install torch

Conda (📥 26M · ⏱️ 28.10.2024):

	conda install -c pytorch pytorch

scikit-learn (🥇53 · ⭐ 61K) - scikit-learn: machine learning in Python. BSD-3

GitHub (👨‍💻 3.3K · 🔀 26K · 📥 1K · 📦 1.1M · 📋 12K - 17% open · ⏱️ 13.02.2025):

	git clone https://github.com/scikit-learn/scikit-learn

PyPi (📥 78M / month · 📦 27K · ⏱️ 10.01.2025):

	pip install scikit-learn

Conda (📥 34M · ⏱️ 10.01.2025):

	conda install -c conda-forge scikit-learn

Keras (🥇48 · ⭐ 63K) - Deep Learning for humans. Apache-2

GitHub (👨‍💻 1.4K · 🔀 20K · 📋 12K - 2% open · ⏱️ 11.02.2025):

	git clone https://github.com/keras-team/keras

PyPi (📥 14M / month · 📦 1.7K · ⏱️ 07.01.2025):

	pip install keras

Conda (📥 3.9M · ⏱️ 10.01.2025):

	conda install -c conda-forge keras

PySpark (🥇45 · ⭐ 41K) - Apache Spark Python API. Apache-2

GitHub (👨‍💻 3.2K · 🔀 28K · ⏱️ 13.02.2025):

	git clone https://github.com/apache/spark

PyPi (📥 33M / month · 📦 1.7K · ⏱️ 20.12.2024):

	pip install pyspark

Conda (📥 3.7M · ⏱️ 21.12.2024):

	conda install -c conda-forge pyspark

jax (🥇45 · ⭐ 31K) - Composable transformations of Python+NumPy programs: differentiate,.. Apache-2

GitHub (👨‍💻 830 · 🔀 2.9K · 📦 37K · 📋 6K - 25% open · ⏱️ 13.02.2025):

	git clone https://github.com/google/jax

PyPi (📥 6.8M / month · 📦 2.1K · ⏱️ 17.01.2025):

	pip install jax

Conda (📥 2.3M · ⏱️ 06.01.2025):

	conda install -c conda-forge jaxlib

XGBoost (🥇45 · ⭐ 27K) - Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or.. Apache-2

GitHub (👨‍💻 660 · 🔀 8.7K · 📥 14K · 📦 130K · 📋 5.4K - 8% open · ⏱️ 13.02.2025):

	git clone https://github.com/dmlc/xgboost

PyPi (📥 24M / month · 📦 2.3K · ⏱️ 06.02.2025):

	pip install xgboost

Conda (📥 5.8M · ⏱️ 07.02.2025):

	conda install -c conda-forge xgboost

PaddlePaddle (🥇45 · ⭐ 22K) - PArallel Distributed Deep LEarning: Machine Learning.. Apache-2

GitHub (👨‍💻 1.3K · 🔀 5.7K · 📥 15K · 📦 7K · 📋 19K - 8% open · ⏱️ 13.02.2025):

	git clone https://github.com/PaddlePaddle/Paddle

PyPi (📥 310K / month · 📦 210 · ⏱️ 23.01.2025):

	pip install paddlepaddle

StatsModels (🥇45 · ⭐ 10K) - Statsmodels: statistical modeling and econometrics in Python. BSD-3

GitHub (👨‍💻 450 · 🔀 3.1K · 📥 35 · 📦 150K · 📋 5.7K - 50% open · ⏱️ 06.02.2025):

	git clone https://github.com/statsmodels/statsmodels

PyPi (📥 16M / month · 📦 4.5K · ⏱️ 03.10.2024):

	pip install statsmodels

Conda (📥 17M · ⏱️ 03.10.2024):

	conda install -c conda-forge statsmodels

pytorch-lightning (🥈43 · ⭐ 29K) - Pretrain, finetune ANY AI model of ANY size on.. Apache-2

GitHub (👨‍💻 990 · 🔀 3.4K · 📥 11K · 📦 42K · 📋 7.2K - 11% open · ⏱️ 13.02.2025):

	git clone https://github.com/Lightning-AI/lightning

PyPi (📥 7.5M / month · 📦 1.5K · ⏱️ 21.12.2024):

	pip install pytorch-lightning

Conda (📥 1.4M · ⏱️ 22.12.2024):

	conda install -c conda-forge pytorch-lightning

LightGBM (🥈41 · ⭐ 17K) - A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT,.. MIT

GitHub (👨‍💻 320 · 🔀 3.8K · 📥 280K · 📦 44K · 📋 3.5K - 10% open · ⏱️ 13.02.2025):

	git clone https://github.com/microsoft/LightGBM

PyPi (📥 8.4M / month · 📦 1.1K · ⏱️ 26.07.2024):

	pip install lightgbm

Conda (📥 3.1M · ⏱️ 26.01.2025):

	conda install -c conda-forge lightgbm

Catboost (🥈41 · ⭐ 8.2K) - A fast, scalable, high performance Gradient Boosting on Decision.. Apache-2

GitHub (👨‍💻 1.3K · 🔀 1.2K · 📥 350K · 📦 16 · 📋 2.4K - 24% open · ⏱️ 13.02.2025):

	git clone https://github.com/catboost/catboost

PyPi (📥 3M / month · 📦 540 · ⏱️ 07.09.2024):

	pip install catboost

Conda (📥 1.9M · ⏱️ 29.01.2025):

	conda install -c conda-forge catboost

PyFlink (🥈40 · ⭐ 25K) - Apache Flink Python API. Apache-2

GitHub (👨‍💻 2K · 🔀 13K · 📦 21 · ⏱️ 13.02.2025):

	git clone https://github.com/apache/flink

PyPi (📥 14M / month · 📦 35 · ⏱️ 12.02.2025):

	pip install apache-flink

Fastai (🥈39 · ⭐ 27K) - The fastai deep learning library. Apache-2

GitHub (👨‍💻 670 · 🔀 7.6K · 📦 21K · 📋 1.8K - 12% open · ⏱️ 14.12.2024):

	git clone https://github.com/fastai/fastai

PyPi (📥 430K / month · 📦 310 · ⏱️ 19.10.2024):

	pip install fastai

einops (🥈37 · ⭐ 8.7K · 📈) - Flexible and powerful tensor operations for readable and reliable code.. MIT

GitHub (👨‍💻 33 · 🔀 360 · 📦 60K · 📋 190 - 17% open · ⏱️ 09.02.2025):

	git clone https://github.com/arogozhnikov/einops

PyPi (📥 7.2M / month · 📦 2.6K · ⏱️ 09.02.2025):

	pip install einops

Conda (📥 330K · ⏱️ 09.02.2025):

	conda install -c conda-forge einops

Flax (🥈37 · ⭐ 6.3K) - Flax is a neural network library for JAX that is designed for.. Apache-2

GitHub (👨‍💻 250 · 🔀 660 · 📥 59 · 📦 12K · 📋 1.1K - 32% open · ⏱️ 13.02.2025):

	git clone https://github.com/google/flax

PyPi (📥 1.3M / month · 📦 540 · ⏱️ 10.02.2025):

	pip install flax

Conda (📥 88K · ⏱️ 20.11.2024):

	conda install -c conda-forge flax

Jina (🥈36 · ⭐ 21K) - Build multimodal AI applications with cloud-native stack. Apache-2

GitHub (👨‍💻 180 · 🔀 2.2K · 📋 1.9K - 0% open · ⏱️ 20.12.2024):

	git clone https://github.com/jina-ai/jina

PyPi (📥 120K / month · 📦 27 · ⏱️ 20.12.2024):

	pip install jina

Conda (📥 86K · ⏱️ 16.06.2023):

	conda install -c conda-forge jina-core

Docker Hub (📥 1.8M · ⭐ 8 · ⏱️ 20.12.2024):

	docker pull jinaai/jina

ivy (🥈36 · ⭐ 14K) - Convert Machine Learning Code Between Frameworks. Apache-2

GitHub (👨‍💻 1.5K · 🔀 5.7K · 📋 17K - 5% open · ⏱️ 02.02.2025):

	git clone https://github.com/unifyai/ivy

PyPi (📥 22K / month · 📦 16 · ⏱️ 22.01.2025):

	pip install ivy

mlpack (🥈34 · ⭐ 5.2K) - mlpack: a fast, header-only C++ machine learning library. BSD-3

GitHub (👨‍💻 330 · 🔀 1.6K · 📋 1.7K - 1% open · ⏱️ 12.02.2025):

	git clone https://github.com/mlpack/mlpack

PyPi (📥 7.5K / month · 📦 6 · ⏱️ 11.12.2024):

	pip install mlpack

Conda (📥 330K · ⏱️ 22.09.2024):

	conda install -c conda-forge mlpack

Ignite (🥈34 · ⭐ 4.6K) - High-level library to help with training and evaluating neural.. BSD-3

GitHub (👨‍💻 770 · 🔀 630 · 📦 3.5K · 📋 1.4K - 10% open · ⏱️ 04.01.2025):

	git clone https://github.com/pytorch/ignite

PyPi (📥 130K / month · 📦 100 · ⏱️ 13.02.2025):

	pip install pytorch-ignite

Conda (📥 220K · ⏱️ 13.08.2024):

	conda install -c pytorch ignite

Thinc (🥈34 · ⭐ 2.8K) - A refreshing functional take on deep learning, compatible with your favorite.. MIT

GitHub (👨‍💻 66 · 🔀 280 · 📥 830 · 📦 61K · 📋 150 - 11% open · ⏱️ 11.12.2024):

	git clone https://github.com/explosion/thinc

PyPi (📥 11M / month · 📦 150 · ⏱️ 13.01.2025):

	pip install thinc

Conda (📥 3.4M · ⏱️ 03.12.2024):

	conda install -c conda-forge thinc

skorch (🥈33 · ⭐ 6K) - A scikit-learn compatible neural network library that wraps.. BSD-3

GitHub (👨‍💻 66 · 🔀 390 · 📦 1.5K · 📋 530 - 12% open · ⏱️ 04.02.2025):

	git clone https://github.com/skorch-dev/skorch

PyPi (📥 110K / month · 📦 94 · ⏱️ 10.01.2025):

	pip install skorch

Conda (📥 790K · ⏱️ 11.01.2025):

	conda install -c conda-forge skorch

Ludwig (🥉32 · ⭐ 11K) - Low-code framework for building custom LLMs, neural networks, and.. Apache-2

GitHub (👨‍💻 160 · 🔀 1.2K · 📦 290 · 📋 1.1K - 4% open · ⏱️ 17.10.2024):

	git clone https://github.com/ludwig-ai/ludwig

PyPi (📥 2.4K / month · 📦 6 · ⏱️ 30.07.2024):

	pip install ludwig

Vowpal Wabbit (🥉32 · ⭐ 8.5K) - Vowpal Wabbit is a machine learning system which pushes the.. BSD-3

GitHub (👨‍💻 340 · 🔀 1.9K · 📦 1 · 📋 1.3K - 10% open · ⏱️ 01.08.2024):

	git clone https://github.com/VowpalWabbit/vowpal_wabbit

PyPi (📥 39K / month · 📦 40 · ⏱️ 08.08.2024):

	pip install vowpalwabbit

Conda (📥 320K · ⏱️ 27.11.2024):

	conda install -c conda-forge vowpalwabbit

Sonnet (🥉31 · ⭐ 9.8K) - TensorFlow-based neural network library. Apache-2

GitHub (👨‍💻 60 · 🔀 1.3K · 📦 1.4K · 📋 190 - 16% open · ⏱️ 30.01.2025):

	git clone https://github.com/deepmind/sonnet

PyPi (📥 21K / month · 📦 19 · ⏱️ 02.01.2024):

	pip install dm-sonnet

Conda (📥 39K · ⏱️ 16.06.2023):

	conda install -c conda-forge sonnet

Haiku (🥉31 · ⭐ 3K) - JAX-based neural network library. Apache-2

GitHub (👨‍💻 85 · 🔀 240 · 📦 2.3K · 📋 250 - 28% open · ⏱️ 29.01.2025):

	git clone https://github.com/deepmind/dm-haiku

PyPi (📥 300K / month · 📦 180 · ⏱️ 16.10.2024):

	pip install dm-haiku

Conda (📥 29K · ⏱️ 23.12.2024):

	conda install -c conda-forge dm-haiku

Geomstats (🥉31 · ⭐ 1.3K) - Computations and statistics on manifolds with geometric structures. MIT

GitHub (👨‍💻 95 · 🔀 250 · 📦 130 · 📋 570 - 36% open · ⏱️ 08.02.2025):

	git clone https://github.com/geomstats/geomstats

PyPi (📥 4.1K / month · 📦 12 · ⏱️ 09.09.2024):

	pip install geomstats

Conda (📥 5.5K · ⏱️ 01.01.2025):

	conda install -c conda-forge geomstats

tensorflow-upstream (🥉31 · ⭐ 690) - TensorFlow ROCm port. Apache-2

GitHub (👨‍💻 4.8K · 🔀 98 · 📥 27 · 📋 390 - 13% open · ⏱️ 12.02.2025):

	git clone https://github.com/ROCmSoftwarePlatform/tensorflow-upstream

PyPi (📥 5.8K / month · 📦 9 · ⏱️ 10.01.2024):

	pip install tensorflow-rocm

Determined (🥉29 · ⭐ 3.1K) - Determined is an open-source machine learning platform.. Apache-2

GitHub (👨‍💻 120 · 🔀 360 · 📥 12K · 📋 450 - 21% open · ⏱️ 05.02.2025):

	git clone https://github.com/determined-ai/determined

PyPi (📥 26K / month · 📦 4 · ⏱️ 22.11.2024):

	pip install determined

NuPIC (🥉28 · ⭐ 6.3K) - Numenta Platform for Intelligent Computing is an implementation of.. MIT

GitHub (👨‍💻 120 · 🔀 1.6K · 📥 18 · 📦 21 · 📋 1.8K - 25% open · ⏱️ 03.12.2024):

	git clone https://github.com/numenta/nupic

PyPi (📥 1.7K / month · ⏱️ 01.09.2016):

	pip install nupic

ktrain (🥉27 · ⭐ 1.2K · 💤) - ktrain is a Python library that makes deep learning and AI.. Apache-2

GitHub (👨‍💻 17 · 🔀 270 · 📦 560 · 📋 500 - 0% open · ⏱️ 09.07.2024):

	git clone https://github.com/amaiya/ktrain

PyPi (📥 7.7K / month · 📦 4 · ⏱️ 19.06.2024):

	pip install ktrain

pyRiemann (🥉27 · ⭐ 660) - Machine learning for multivariate data through the Riemannian.. BSD-3

GitHub (👨‍💻 36 · 🔀 160 · 📦 430 · 📋 110 - 2% open · ⏱️ 12.02.2025):

	git clone https://github.com/pyRiemann/pyRiemann

PyPi (📥 40K / month · 📦 28 · ⏱️ 12.02.2025):

	pip install pyriemann

Conda (📥 11K · ⏱️ 12.02.2025):

	conda install -c conda-forge pyriemann

Neural Network Libraries (🥉26 · ⭐ 2.7K) - Neural Network Libraries. Apache-2

GitHub (👨‍💻 76 · 🔀 330 · 📥 1K · 📋 95 - 36% open · ⏱️ 15.11.2024):

	git clone https://github.com/sony/nnabla

PyPi (📥 19K / month · 📦 44 · ⏱️ 29.05.2024):

	pip install nnabla

Towhee (🥉24 · ⭐ 3.3K) - Towhee is a framework that is dedicated to making neural data.. Apache-2

GitHub (👨‍💻 38 · 🔀 250 · 📥 2.7K · 📋 670 - 0% open · ⏱️ 18.10.2024):

	git clone https://github.com/towhee-io/towhee

PyPi (📥 17K / month · ⏱️ 04.12.2023):

	pip install towhee

Neural Tangents (🥉24 · ⭐ 2.3K · 💤) - Fast and Easy Infinite Neural Networks in Python. Apache-2

GitHub (👨‍💻 30 · 🔀 240 · 📥 540 · 📦 120 · 📋 160 - 38% open · ⏱️ 01.03.2024):

	git clone https://github.com/google/neural-tangents

PyPi (📥 3.5K / month · 📦 1 · ⏱️ 11.12.2023):

	pip install neural-tangents

Runhouse (🥉24 · ⭐ 1K) - The Runhouse Python client. Distribute and run AI workloads magically.. Apache-2

GitHub (👨‍💻 16 · 🔀 38 · 📥 69 · 📋 51 - 17% open · ⏱️ 13.02.2025):

	git clone https://github.com/run-house/runhouse

PyPi (📥 37K / month · 📦 1 · ⏱️ 12.02.2025):

	pip install runhouse

fklearn (🥉23 · ⭐ 1.5K) - fklearn: Functional Machine Learning. Apache-2

GitHub (👨‍💻 56 · 🔀 170 · 📦 16 · 📋 65 - 61% open · ⏱️ 14.08.2024):

	git clone https://github.com/nubank/fklearn

PyPi (📥 2.9K / month · ⏱️ 14.08.2024):

	pip install fklearn

ThunderSVM (🥉22 · ⭐ 1.6K · 💤) - ThunderSVM: A Fast SVM Library on GPUs and CPUs. Apache-2

GitHub (👨‍💻 37 · 🔀 220 · 📥 2.9K · 📋 230 - 35% open · ⏱️ 01.04.2024):

	git clone https://github.com/Xtra-Computing/thundersvm

PyPi (📥 1.3K / month · ⏱️ 13.03.2020):

	pip install thundersvm

mace (🥉21 · ⭐ 5K · 💤) - MACE is a deep learning inference framework optimized for mobile.. Apache-2

GitHub (👨‍💻 69 · 🔀 820 · 📥 1.5K · 📋 680 - 8% open · ⏱️ 11.03.2024):

	git clone https://github.com/XiaoMi/mace

chefboost (🥉21 · ⭐ 470) - A Lightweight Decision Tree Framework supporting regular algorithms:.. MIT

GitHub (👨‍💻 7 · 🔀 100 · 📦 66 · ⏱️ 30.10.2024):
```
	git clone https://github.com/serengil/chefboost
```
PyPi (📥 6.6K / month · ⏱️ 30.10.2024):
```
	pip install chefboost
```

NeoML (🥉19 · ⭐ 770) - Machine learning framework for both deep learning and traditional.. Apache-2

GitHub (👨‍💻 40 · 🔀 130 · 📦 2 · 📋 91 - 40% open · ⏱️ 30.09.2024):

	git clone https://github.com/neoml-lib/neoml

PyPi (📥 960 / month · ⏱️ 26.12.2023):

	pip install neoml

Show 22 hidden projects...

MXNet (🥈38 · ⭐ 21K · 💀) - Lightweight, Portable, Flexible Distributed/Mobile Deep.. Apache-2

dlib (🥈38 · ⭐ 14K) - A toolkit for making real world machine learning and data analysis.. ❗️BSL-1.0

Theano (🥈37 · ⭐ 9.9K · 💀) - Theano was a Python library that allows you to define, optimize, and.. BSD-3

MindsDB (🥈33 · ⭐ 27K) - AGIs query engine - Platform for building AI that can.. ❗️MulanPSL-1.0

Turi Create (🥈33 · ⭐ 11K · 💀) - Turi Create simplifies the development of custom machine.. BSD-3

tensorpack (🥈33 · ⭐ 6.3K · 💀) - A Neural Net Training Interface on TensorFlow, with.. Apache-2

Chainer (🥈33 · ⭐ 5.9K · 💀) - A flexible framework of neural networks for deep learning. MIT

TFlearn (🥉31 · ⭐ 9.6K · 💀) - Deep learning library featuring a higher-level API for TensorFlow. MIT

dyNET (🥉31 · ⭐ 3.4K · 💀) - DyNet: The Dynamic Neural Network Toolkit. Apache-2

CNTK (🥉29 · ⭐ 18K · 💀) - Microsoft Cognitive Toolkit (CNTK), an open source deep-learning toolkit. MIT

Lasagne (🥉28 · ⭐ 3.9K · 💀) - Lightweight library to build and train neural networks in Theano. MIT

EvaDB (🥉27 · ⭐ 2.7K · 💀) - Database system for AI-powered apps. Apache-2

SHOGUN (🥉26 · ⭐ 3K · 💀) - Unified and efficient Machine Learning. BSD-3

xLearn (🥉24 · ⭐ 3.1K · 💀) - High performance, easy-to-use, and scalable machine learning (ML).. Apache-2

NeuPy (🥉24 · ⭐ 740 · 💀) - NeuPy is a Tensorflow based python library for prototyping and building.. MIT

neon (🥉23 · ⭐ 3.9K · 💀) - Intel Nervana reference deep learning framework committed to best.. Apache-2

Objax (🥉21 · ⭐ 770 · 💀) - Objax is a machine learning framework that provides an Object.. Apache-2

Torchbearer (🥉21 · ⭐ 640 · 💀) - torchbearer: A model fitting library for PyTorch. MIT

elegy (🥉19 · ⭐ 470 · 💀) - A High Level API for Deep Learning in JAX. MIT

ThunderGBM (🥉18 · ⭐ 690 · 💀) - ThunderGBM: Fast GBDTs and Random Forests on GPUs. Apache-2

StarSpace (🥉16 · ⭐ 4K · 💀) - Learning embeddings for classification, retrieval and ranking. MIT

nanodl (🥉14 · ⭐ 280) - A Jax-based library for designing and training transformer models from.. MIT

Data Visualization

General-purpose and task-specific data visualization libraries.

Matplotlib (🥇49 · ⭐ 21K) - matplotlib: plotting with Python. ❗Unlicensed

GitHub (👨‍💻 1.8K · 🔀 7.7K · 📦 1.5M · 📋 11K - 14% open · ⏱️ 13.02.2025):
```
	git clone https://github.com/matplotlib/matplotlib
```
PyPi (📥 72M / month · 📦 53K · ⏱️ 14.12.2024):
```
	pip install matplotlib
```

Conda (📥 28M · ⏱️ 16.12.2024):

	conda install -c conda-forge matplotlib

Plotly (🥇46 · ⭐ 17K) - The interactive graphing library for Python This project now includes.. MIT

GitHub (👨‍💻 280 · 🔀 2.6K · 📥 26 · 📦 360K · 📋 3.1K - 19% open · ⏱️ 13.02.2025):

	git clone https://github.com/plotly/plotly.py

PyPi (📥 20M / month · 📦 7.3K · ⏱️ 28.01.2025):

	pip install plotly

Conda (📥 8.3M · ⏱️ 31.01.2025):

	conda install -c conda-forge plotly

npm (📥 68K / month · 📦 9 · ⏱️ 12.01.2021):

	npm install plotlywidget

Bokeh (🥇45 · ⭐ 20K) - Interactive Data Visualization in the browser, from Python. BSD-3

GitHub (👨‍💻 710 · 🔀 4.2K · 📦 98K · 📋 7.9K - 10% open · ⏱️ 12.02.2025):

	git clone https://github.com/bokeh/bokeh

PyPi (📥 4.1M / month · 📦 1.9K · ⏱️ 12.02.2025):

	pip install bokeh

Conda (📥 16M · ⏱️ 06.02.2025):

	conda install -c conda-forge bokeh

dash (🥇43 · ⭐ 22K) - Data Apps & Dashboards for Python. No JavaScript Required. MIT

GitHub (👨‍💻 170 · 🔀 2.1K · 📥 88 · 📦 77K · 📋 1.9K - 27% open · ⏱️ 12.02.2025):

	git clone https://github.com/plotly/dash

PyPi (📥 4.9M / month · 📦 1.5K · ⏱️ 28.01.2025):

	pip install dash

Conda (📥 1.7M · ⏱️ 24.01.2025):

	conda install -c conda-forge dash

Altair (🥇43 · ⭐ 9.6K) - Declarative visualization library for Python. BSD-3

GitHub (👨‍💻 170 · 🔀 790 · 📥 220 · 📦 200K · 📋 2.1K - 6% open · ⏱️ 10.02.2025):

	git clone https://github.com/altair-viz/altair

PyPi (📥 25M / month · 📦 920 · ⏱️ 23.11.2024):

	pip install altair

Conda (📥 2.6M · ⏱️ 15.12.2024):

	conda install -c conda-forge altair

Seaborn (🥇42 · ⭐ 13K) - Statistical data visualization in Python. BSD-3

GitHub (👨‍💻 220 · 🔀 1.9K · 📥 470 · 📦 560K · 📋 2.6K - 6% open · ⏱️ 26.01.2025):

	git clone https://github.com/mwaskom/seaborn

PyPi (📥 22M / month · 📦 11K · ⏱️ 25.01.2024):

	pip install seaborn

Conda (📥 12M · ⏱️ 09.12.2024):

	conda install -c conda-forge seaborn

FiftyOne (🥈38 · ⭐ 9.2K) - Visualize, create, and debug image and video datasets.. Apache-2

GitHub (👨‍💻 150 · 🔀 590 · 📦 820 · 📋 1.6K - 32% open · ⏱️ 13.02.2025):

	git clone https://github.com/voxel51/fiftyone

PyPi (📥 67K / month · 📦 25 · ⏱️ 24.01.2025):

	pip install fiftyone

PyVista (🥈38 · ⭐ 2.9K) - 3D plotting and mesh analysis through a streamlined interface for.. MIT

GitHub (👨‍💻 170 · 🔀 520 · 📥 870 · 📦 4.1K · 📋 1.8K - 36% open · ⏱️ 12.02.2025):

	git clone https://github.com/pyvista/pyvista

PyPi (📥 420K / month · 📦 580 · ⏱️ 27.11.2024):

	pip install pyvista

Conda (📥 630K · ⏱️ 15.12.2024):

	conda install -c conda-forge pyvista

pandas-profiling (🥈37 · ⭐ 13K) - 1 Line of code data quality profiling & exploratory.. MIT

GitHub (👨‍💻 130 · 🔀 1.7K · 📥 260 · 📦 5.4K · 📋 820 - 28% open · ⏱️ 24.01.2025):

	git clone https://github.com/ydataai/pandas-profiling

PyPi (📥 430K / month · 📦 180 · ⏱️ 03.02.2023):

	pip install pandas-profiling

Conda (📥 490K · ⏱️ 16.06.2023):

	conda install -c conda-forge pandas-profiling

HoloViews (🥈37 · ⭐ 2.7K) - With Holoviews, your data visualizes itself. BSD-3

GitHub (👨‍💻 150 · 🔀 400 · 📦 14K · 📋 3.4K - 33% open · ⏱️ 13.02.2025):

	git clone https://github.com/holoviz/holoviews

PyPi (📥 450K / month · 📦 400 · ⏱️ 11.11.2024):

	pip install holoviews

Conda (📥 2M · ⏱️ 13.12.2024):

	conda install -c conda-forge holoviews

npm (📥 360 / month · 📦 5 · ⏱️ 14.01.2025):

	npm install @pyviz/jupyterlab_pyviz

pyecharts (🥈36 · ⭐ 15K) - Python Echarts Plotting Library. MIT

GitHub (👨‍💻 45 · 🔀 2.9K · 📥 72 · 📦 4.9K · ⏱️ 26.01.2025):

	git clone https://github.com/pyecharts/pyecharts

PyPi (📥 300K / month · 📦 220 · ⏱️ 24.01.2025):

	pip install pyecharts

PyQtGraph (🥈36 · ⭐ 4K) - Fast data visualization and GUI tools for scientific / engineering.. MIT

GitHub (👨‍💻 300 · 🔀 1.1K · 📦 11K · 📋 1.3K - 32% open · ⏱️ 06.02.2025):

	git clone https://github.com/pyqtgraph/pyqtgraph

PyPi (📥 360K / month · 📦 1K · ⏱️ 29.04.2024):

	pip install pyqtgraph

Conda (📥 650K · ⏱️ 11.12.2024):

	conda install -c conda-forge pyqtgraph

plotnine (🥈35 · ⭐ 4.1K) - A Grammar of Graphics for Python. MIT

GitHub (👨‍💻 110 · 🔀 230 · 📦 10K · 📋 700 - 11% open · ⏱️ 02.01.2025):

	git clone https://github.com/has2k1/plotnine

PyPi (📥 3M / month · 📦 340 · ⏱️ 02.01.2025):

	pip install plotnine

Conda (📥 440K · ⏱️ 02.01.2025):

	conda install -c conda-forge plotnine

Graphviz (🥈35 · ⭐ 1.7K · 💤) - Simple Python interface for Graphviz. MIT

GitHub (👨‍💻 23 · 🔀 210 · 📦 82K · 📋 180 - 6% open · ⏱️ 13.05.2024):

	git clone https://github.com/xflr6/graphviz

PyPi (📥 16M / month · 📦 2.6K · ⏱️ 21.03.2024):

	pip install graphviz

Conda (📥 52K · ⏱️ 08.04.2024):

	conda install -c anaconda python-graphviz

cartopy (🥈35 · ⭐ 1.5K) - Cartopy - a cartographic python library with matplotlib support. BSD-3

GitHub (👨‍💻 130 · 🔀 370 · 📦 6.6K · 📋 1.3K - 23% open · ⏱️ 24.01.2025):

	git clone https://github.com/SciTools/cartopy

PyPi (📥 420K / month · 📦 720 · ⏱️ 08.10.2024):

	pip install cartopy

Conda (📥 4.5M · ⏱️ 07.10.2024):

	conda install -c conda-forge cartopy

wordcloud (🥈34 · ⭐ 10K) - A little word cloud generator in Python. MIT

GitHub (👨‍💻 72 · 🔀 2.3K · 📦 21 · 📋 550 - 23% open · ⏱️ 10.11.2024):

	git clone https://github.com/amueller/word_cloud

PyPi (📥 1.7M / month · 📦 550 · ⏱️ 10.11.2024):

	pip install wordcloud

Conda (📥 630K · ⏱️ 02.12.2024):

	conda install -c conda-forge wordcloud

Perspective (🥈34 · ⭐ 8.8K) - A data visualization and analytics component, especially.. Apache-2

GitHub (👨‍💻 98 · 🔀 1.2K · 📥 8.7K · 📦 160 · 📋 860 - 12% open · ⏱️ 10.02.2025):

	git clone https://github.com/finos/perspective

PyPi (📥 20K / month · 📦 28 · ⏱️ 10.02.2025):

	pip install perspective-python

Conda (📥 1.8M · ⏱️ 10.02.2025):

	conda install -c conda-forge perspective

npm (📥 760 / month · 📦 6 · ⏱️ 10.02.2025):

	npm install @finos/perspective-jupyterlab

datashader (🥈34 · ⭐ 3.4K) - Quickly and accurately render even the largest data. BSD-3

GitHub (👨‍💻 61 · 🔀 370 · 📦 5.5K · 📋 600 - 23% open · ⏱️ 13.02.2025):

	git clone https://github.com/holoviz/datashader

PyPi (📥 170K / month · 📦 230 · ⏱️ 29.01.2025):

	pip install datashader

Conda (📥 1.3M · ⏱️ 30.01.2025):

	conda install -c conda-forge datashader

UMAP (🥈33 · ⭐ 7.6K) - Uniform Manifold Approximation and Projection. BSD-3

GitHub (👨‍💻 140 · 🔀 820 · 📦 1 · 📋 830 - 58% open · ⏱️ 29.11.2024):

	git clone https://github.com/lmcinnes/umap

PyPi (📥 1.8M / month · 📦 1.1K · ⏱️ 28.10.2024):

	pip install umap-learn

Conda (📥 2.8M · ⏱️ 11.02.2025):

	conda install -c conda-forge umap-learn

VisPy (🥈33 · ⭐ 3.4K) - High-performance interactive 2D/3D data visualization library. BSD-3

GitHub (👨‍💻 200 · 🔀 620 · 📦 1.8K · 📋 1.5K - 24% open · ⏱️ 22.01.2025):

	git clone https://github.com/vispy/vispy

PyPi (📥 110K / month · 📦 170 · ⏱️ 17.06.2024):

	pip install vispy

Conda (📥 740K · ⏱️ 04.09.2024):

	conda install -c conda-forge vispy

npm (📥 33 / month · 📦 3 · ⏱️ 15.03.2020):

	npm install vispy

hvPlot (🥈32 · ⭐ 1.2K) - A high-level plotting API for pandas, dask, xarray, and networkx built.. BSD-3

GitHub (👨‍💻 51 · 🔀 110 · 📦 6.5K · 📋 840 - 43% open · ⏱️ 11.02.2025):

	git clone https://github.com/holoviz/hvplot

PyPi (📥 200K / month · 📦 220 · ⏱️ 16.12.2024):

	pip install hvplot

Conda (📥 710K · ⏱️ 16.12.2024):

	conda install -c conda-forge hvplot

mpld3 (🥈31 · ⭐ 2.4K) - An interactive data visualization tool which brings matplotlib graphics to.. BSD-3

GitHub (👨‍💻 53 · 🔀 360 · 📦 7K · 📋 370 - 59% open · ⏱️ 30.10.2024):

	git clone https://github.com/mpld3/mpld3

PyPi (📥 350K / month · 📦 150 · ⏱️ 23.12.2023):

	pip install mpld3

Conda (📥 220K · ⏱️ 19.12.2024):

	conda install -c conda-forge mpld3

npm (📥 940 / month · 📦 9 · ⏱️ 23.12.2023):

	npm install mpld3

lets-plot (🥈31 · ⭐ 1.6K) - Multiplatform plotting library based on the Grammar of Graphics. MIT

GitHub (👨‍💻 21 · 🔀 53 · 📥 3.2K · 📦 160 · 📋 670 - 24% open · ⏱️ 12.02.2025):

	git clone https://github.com/JetBrains/lets-plot

PyPi (📥 31K / month · 📦 15 · ⏱️ 17.12.2024):

	pip install lets-plot

D-Tale (🥉30 · ⭐ 4.8K) - Visualizer for pandas data structures. ❗️LGPL-2.1

GitHub (👨‍💻 30 · 🔀 410 · 📦 1.4K · 📋 600 - 10% open · ⏱️ 13.12.2024):

	git clone https://github.com/man-group/dtale

PyPi (📥 98K / month · 📦 48 · ⏱️ 13.12.2024):

	pip install dtale

Conda (📥 390K · ⏱️ 13.12.2024):

	conda install -c conda-forge dtale

bqplot (🥉30 · ⭐ 3.6K) - Plotting library for IPython/Jupyter notebooks. Apache-2

GitHub (👨‍💻 65 · 🔀 470 · 📦 61 · 📋 640 - 42% open · ⏱️ 22.10.2024):

	git clone https://github.com/bqplot/bqplot

PyPi (📥 170K / month · 📦 110 · ⏱️ 24.12.2024):

	pip install bqplot

Conda (📥 1.5M · ⏱️ 16.12.2024):

	conda install -c conda-forge bqplot

npm (📥 2.2K / month · 📦 21 · ⏱️ 24.12.2024):

	npm install bqplot

AutoViz (🥉26 · ⭐ 1.8K · 💤) - Automatically Visualize any dataset, any size with a single line.. Apache-2

GitHub (👨‍💻 17 · 🔀 200 · 📦 820 · 📋 98 - 2% open · ⏱️ 10.06.2024):

	git clone https://github.com/AutoViML/AutoViz

PyPi (📥 18K / month · 📦 11 · ⏱️ 10.06.2024):

	pip install autoviz

Conda (📥 77K · ⏱️ 17.01.2025):

	conda install -c conda-forge autoviz

openTSNE (🥉26 · ⭐ 1.5K) - Extensible, parallel implementations of t-SNE. BSD-3

GitHub (👨‍💻 13 · 🔀 170 · 📦 970 · 📋 140 - 7% open · ⏱️ 24.10.2024):

	git clone https://github.com/pavlin-policar/openTSNE

PyPi (📥 44K / month · 📦 47 · ⏱️ 13.08.2024):

	pip install opentsne

Conda (📥 400K · ⏱️ 16.11.2024):

	conda install -c conda-forge opentsne

data-validation (🥉26 · ⭐ 770) - Library for exploring and validating machine learning.. Apache-2

GitHub (👨‍💻 27 · 🔀 170 · 📥 950 · 📋 180 - 21% open · ⏱️ 30.01.2025):

	git clone https://github.com/tensorflow/data-validation

PyPi (📥 170K / month · 📦 31 · ⏱️ 15.10.2024):

	pip install tensorflow-data-validation

Chartify (🥉25 · ⭐ 3.6K) - Python library that makes it easy for data scientists to create.. Apache-2

GitHub (👨‍💻 27 · 🔀 320 · 📦 80 · 📋 83 - 61% open · ⏱️ 16.10.2024):

	git clone https://github.com/spotify/chartify

PyPi (📥 1.9K / month · 📦 9 · ⏱️ 16.10.2024):

	pip install chartify

Conda (📥 36K · ⏱️ 16.06.2023):

	conda install -c conda-forge chartify

HyperTools (🥉25 · ⭐ 1.8K · 💤) - A Python toolbox for gaining geometric insights into high-.. MIT

GitHub (👨‍💻 22 · 🔀 160 · 📥 68 · 📦 500 · 📋 200 - 34% open · ⏱️ 19.03.2024):

	git clone https://github.com/ContextLab/hypertools

PyPi (📥 570 / month · 📦 2 · ⏱️ 12.02.2022):

	pip install hypertools

Plotly-Resampler (🥉25 · ⭐ 1.1K) - Visualize large time series data with plotly.py. MIT

GitHub (👨‍💻 14 · 🔀 71 · 📦 1.7K · 📋 180 - 32% open · ⏱️ 15.12.2024):

	git clone https://github.com/predict-idlab/plotly-resampler

PyPi (📥 410K / month · 📦 24 · ⏱️ 27.03.2024):

	pip install plotly-resampler

Conda (📥 94K · ⏱️ 05.12.2024):

	conda install -c conda-forge plotly-resampler

python-ternary (🥉25 · ⭐ 750 · 💤) - Ternary plotting library for python with matplotlib. MIT

GitHub (👨‍💻 28 · 🔀 160 · 📥 35 · 📦 200 · 📋 140 - 24% open · ⏱️ 12.06.2024):

	git clone https://github.com/marcharper/python-ternary

PyPi (📥 21K / month · 📦 32 · ⏱️ 17.02.2021):

	pip install python-ternary

Conda (📥 97K · ⏱️ 03.01.2025):

	conda install -c conda-forge python-ternary

Multicore-TSNE (🥉23 · ⭐ 1.9K · 💤) - Parallel t-SNE implementation with Python and Torch.. BSD-3

GitHub (👨‍💻 18 · 🔀 230 · 📦 490 · 📋 69 - 65% open · ⏱️ 06.02.2024):

	git clone https://github.com/DmitryUlyanov/Multicore-TSNE

PyPi (📥 1.5K / month · 📦 22 · ⏱️ 09.01.2019):

	pip install MulticoreTSNE

Conda (📥 67K · ⏱️ 11.10.2023):

	conda install -c conda-forge multicore-tsne

vega (🥉23 · ⭐ 380) - IPython/Jupyter notebook module for Vega and Vega-Lite. BSD-3

GitHub (👨‍💻 15 · 🔀 65 · 📦 4 · 📋 110 - 14% open · ⏱️ 01.01.2025):

	git clone https://github.com/vega/ipyvega

PyPi (📥 11K / month · 📦 17 · ⏱️ 25.09.2024):

	pip install vega

Conda (📥 700K · ⏱️ 25.09.2024):

	conda install -c conda-forge vega

PyWaffle (🥉22 · ⭐ 590 · 💤) - Make Waffle Charts in Python. MIT

GitHub (👨‍💻 6 · 🔀 110 · 📦 460 · 📋 22 - 27% open · ⏱️ 16.06.2024):

	git clone https://github.com/gyli/PyWaffle

PyPi (📥 12K / month · 📦 6 · ⏱️ 16.06.2024):

	pip install pywaffle

Conda (📥 15K · ⏱️ 16.01.2025):

	conda install -c conda-forge pywaffle

vegafusion (🥉22 · ⭐ 350) - Serverside scaling for Vega and Altair visualizations. BSD-3

GitHub (👨‍💻 5 · 🔀 18 · 📥 9.8K · 📋 140 - 36% open · ⏱️ 25.11.2024):

	git clone https://github.com/vegafusion/vegafusion

PyPi (📥 3K / month · 📦 2 · ⏱️ 09.05.2024):

	pip install vegafusion-jupyter

Conda (📥 380K · ⏱️ 31.10.2024):

	conda install -c conda-forge vegafusion-python-embed

npm (📥 98 / month · 📦 3 · ⏱️ 09.05.2024):

	npm install vegafusion-jupyter

Popmon (🥉21 · ⭐ 500) - Monitor the stability of a Pandas or Spark dataframe. MIT

GitHub (👨‍💻 19 · 🔀 36 · 📥 260 · 📦 22 · 📋 57 - 28% open · ⏱️ 24.01.2025):

	git clone https://github.com/ing-bank/popmon

PyPi (📥 7.7K / month · 📦 4 · ⏱️ 24.01.2025):

	pip install popmon

ivis (🥉19 · ⭐ 330) - Dimensionality reduction in very large datasets using Siamese.. Apache-2

GitHub (👨‍💻 10 · 🔀 44 · 📦 36 · 📋 60 - 5% open · ⏱️ 29.09.2024):

	git clone https://github.com/beringresearch/ivis

PyPi (📥 1.6K / month · 📦 2 · ⏱️ 13.06.2024):

	pip install ivis

animatplot (🥉18 · ⭐ 410) - A python package for animating plots build on matplotlib. MIT

GitHub (👨‍💻 6 · 🔀 38 · 📦 72 · 📋 37 - 45% open · ⏱️ 29.08.2024):

	git clone https://github.com/t-makaro/animatplot

PyPi (📥 450 / month · 📦 4 · ⏱️ 29.08.2024):

	pip install animatplot

Conda (📥 16K · ⏱️ 01.09.2024):

	conda install -c conda-forge animatplot

Show 16 hidden projects...

missingno (🥉29 · ⭐ 4K · 💀) - Missing data visualization module for Python. MIT

Cufflinks (🥉29 · ⭐ 3K · 💀) - Productivity Tools for Plotly + Pandas. MIT

Facets Overview (🥉27 · ⭐ 7.4K · 💀) - Visualizations for machine learning datasets. Apache-2

Sweetviz (🥉27 · ⭐ 3K · 💀) - Visualize and compare datasets, target values and associations, with.. MIT

pythreejs (🥉27 · ⭐ 960 · 💀) - A Jupyter - Three.js bridge. BSD-3

ridgeplot (🥉25 · ⭐ 220) - Beautiful ridgeline plots in Python. MIT

HiPlot (🥉24 · ⭐ 2.8K · 💀) - HiPlot makes understanding high dimensional data easy. MIT

PandasGUI (🥉23 · ⭐ 3.2K · 💀) - A GUI for Pandas DataFrames. ❗️MIT-0

Pandas-Bokeh (🥉23 · ⭐ 880 · 💀) - Bokeh Plotting Backend for Pandas and GeoPandas. MIT

pivottablejs (🥉23 · ⭐ 700 · 💀) - Dragndrop Pivot Tables and Charts for Jupyter/IPython.. MIT

joypy (🥉22 · ⭐ 580 · 💀) - Joyplots in Python with matplotlib & pandas. MIT

PDPbox (🥉21 · ⭐ 850 · 💀) - python partial dependence plot toolbox. MIT

pdvega (🥉16 · ⭐ 340 · 💀) - Interactive plotting for Pandas using Vega-Lite. MIT

data-describe (🥉15 · ⭐ 300 · 💀) - datadescribe: Pythonic EDA Accelerator for Data Science. Apache-2

nx-altair (🥉15 · ⭐ 220 · 💀) - Draw interactive NetworkX graphs with Altair. MIT

nptsne (🥉13 · ⭐ 33 · 💀) - nptsne is a numpy compatible python binary package that offers a.. Apache-2

Text Data & NLP

Libraries for processing, cleaning, manipulating, and analyzing text data as well as libraries for NLP tasks such as language detection, fuzzy matching, classification, seq2seq learning, conversational AI, keyword extraction, and translation.

transformers (🥇52 · ⭐ 140K) - Transformers: State-of-the-art Machine Learning for.. Apache-2

GitHub (👨‍💻 3.1K · 🔀 28K · 📦 290K · 📋 17K - 9% open · ⏱️ 13.02.2025):
```
	git clone https://github.com/huggingface/transformers
```
PyPi (📥 43M / month · 📦 7.6K · ⏱️ 07.02.2025):
```
	pip install transformers
```

Conda (📥 2.5M · ⏱️ 07.02.2025):

	conda install -c conda-forge transformers

spaCy (🥇44 · ⭐ 31K) - Industrial-strength Natural Language Processing (NLP) in Python. MIT

GitHub (👨‍💻 760 · 🔀 4.4K · 📥 1.5K · 📦 110K · 📋 5.7K - 3% open · ⏱️ 03.02.2025):

	git clone https://github.com/explosion/spaCy

PyPi (📥 12M / month · 📦 2.9K · ⏱️ 14.01.2025):

	pip install spacy

Conda (📥 5.3M · ⏱️ 24.11.2024):

	conda install -c conda-forge spacy

litellm (🥇44 · ⭐ 17K) - Python SDK, Proxy Server (LLM Gateway) to call 100+.. MIT o t h e r s

GitHub (👨‍💻 400 · 🔀 2.1K · 📥 440 · 📦 7.4K · 📋 4.6K - 26% open · ⏱️ 13.02.2025):

	git clone https://github.com/BerriAI/litellm

PyPi (📥 5.5M / month · 📦 800 · ⏱️ 12.02.2025):

	pip install litellm

nltk (🥇44 · ⭐ 14K) - Suite of libraries and programs for symbolic and statistical natural.. Apache-2

GitHub (👨‍💻 460 · 🔀 2.9K · 📦 350K · 📋 1.9K - 14% open · ⏱️ 13.02.2025):

	git clone https://github.com/nltk/nltk

PyPi (📥 26M / month · 📦 4.7K · ⏱️ 18.08.2024):

	pip install nltk

Conda (📥 3M · ⏱️ 16.12.2024):

	conda install -c conda-forge nltk

sentence-transformers (🥇42 · ⭐ 16K) - State-of-the-Art Text Embeddings. Apache-2

GitHub (👨‍💻 210 · 🔀 2.5K · 📦 71K · 📋 2.3K - 52% open · ⏱️ 11.02.2025):

	git clone https://github.com/UKPLab/sentence-transformers

PyPi (📥 7.5M / month · 📦 2.1K · ⏱️ 29.01.2025):

	pip install sentence-transformers

Conda (📥 540K · ⏱️ 30.01.2025):

	conda install -c conda-forge sentence-transformers

flair (🥇41 · ⭐ 14K) - A very simple framework for state-of-the-art Natural Language Processing.. MIT

GitHub (👨‍💻 280 · 🔀 2.1K · 📦 3.8K · 📋 2.4K - 4% open · ⏱️ 05.02.2025):

	git clone https://github.com/flairNLP/flair

PyPi (📥 140K / month · 📦 150 · ⏱️ 05.02.2025):

	pip install flair

Conda (📥 39K · ⏱️ 26.12.2024):

	conda install -c conda-forge python-flair

Tokenizers (🥇40 · ⭐ 9.4K) - Fast State-of-the-Art Tokenizers optimized for Research and.. Apache-2

GitHub (👨‍💻 110 · 🔀 820 · 📥 73 · 📦 140K · 📋 1K - 5% open · ⏱️ 11.02.2025):

	git clone https://github.com/huggingface/tokenizers

PyPi (📥 37M / month · 📦 1.1K · ⏱️ 27.11.2024):

	pip install tokenizers

Conda (📥 2.6M · ⏱️ 27.11.2024):

	conda install -c conda-forge tokenizers

Rasa (🥇39 · ⭐ 19K) - Open source machine learning framework to automate text- and voice-.. Apache-2

GitHub (👨‍💻 590 · 🔀 4.7K · 📦 4.8K · 📋 6.8K - 2% open · ⏱️ 14.01.2025):

	git clone https://github.com/RasaHQ/rasa

PyPi (📥 280K / month · 📦 60 · ⏱️ 14.01.2025):

	pip install rasa

gensim (🥇38 · ⭐ 16K) - Topic Modelling for Humans. ❗️LGPL-2.1

GitHub (👨‍💻 460 · 🔀 4.4K · 📥 5.8K · 📦 71K · 📋 1.9K - 20% open · ⏱️ 05.12.2024):

	git clone https://github.com/RaRe-Technologies/gensim

PyPi (📥 4.5M / month · 📦 1.4K · ⏱️ 19.07.2024):

	pip install gensim

Conda (📥 1.5M · ⏱️ 03.09.2024):

	conda install -c conda-forge gensim

ChatterBot (🥇38 · ⭐ 14K · 📈) - ChatterBot is a machine learning, conversational dialog engine.. BSD-3

GitHub (👨‍💻 110 · 🔀 4.5K · 📦 6.1K · 📋 1.7K - 21% open · ⏱️ 12.02.2025):

	git clone https://github.com/gunthercox/ChatterBot

PyPi (📥 34K / month · 📦 18 · ⏱️ 09.02.2025):

	pip install chatterbot

TextBlob (🥇38 · ⭐ 9.2K) - Simple, Pythonic, text processing--Sentiment analysis, part-of-speech.. MIT

GitHub (👨‍💻 37 · 🔀 1.2K · 📥 120 · 📦 49K · 📋 280 - 34% open · ⏱️ 11.02.2025):

	git clone https://github.com/sloria/TextBlob

PyPi (📥 1.2M / month · 📦 400 · ⏱️ 13.01.2025):

	pip install textblob

Conda (📥 280K · ⏱️ 16.06.2023):

	conda install -c conda-forge textblob

fairseq (🥈37 · ⭐ 31K) - Facebook AI Research Sequence-to-Sequence Toolkit written in Python. MIT

GitHub (👨‍💻 430 · 🔀 6.4K · 📥 380 · 📦 4K · 📋 4.3K - 29% open · ⏱️ 18.10.2024):

	git clone https://github.com/facebookresearch/fairseq

PyPi (📥 120K / month · 📦 120 · ⏱️ 27.06.2022):

	pip install fairseq

Conda (📥 130K · ⏱️ 24.11.2024):

	conda install -c conda-forge fairseq

NeMo (🥈37 · ⭐ 13K) - A scalable generative AI framework built for researchers and.. Apache-2

GitHub (👨‍💻 390 · 🔀 2.7K · 📥 330K · 📦 21 · 📋 2.5K - 5% open · ⏱️ 13.02.2025):

	git clone https://github.com/NVIDIA/NeMo

PyPi (📥 82K / month · 📦 14 · ⏱️ 04.02.2025):

	pip install nemo-toolkit

sentencepiece (🥈37 · ⭐ 11K) - Unsupervised text tokenizer for Neural Network-based text.. Apache-2

GitHub (👨‍💻 89 · 🔀 1.2K · 📥 51K · 📦 94K · 📋 760 - 5% open · ⏱️ 18.08.2024):

	git clone https://github.com/google/sentencepiece

PyPi (📥 22M / month · 📦 1.7K · ⏱️ 19.02.2024):

	pip install sentencepiece

Conda (📥 1.4M · ⏱️ 31.12.2024):

	conda install -c conda-forge sentencepiece

TensorFlow Text (🥈36 · ⭐ 1.2K) - Making text a first-class citizen in TensorFlow. Apache-2

GitHub (👨‍💻 170 · 🔀 350 · 📦 8K · 📋 360 - 52% open · ⏱️ 12.02.2025):

	git clone https://github.com/tensorflow/text

PyPi (📥 7.1M / month · 📦 220 · ⏱️ 16.12.2024):

	pip install tensorflow-text

haystack (🥈35 · ⭐ 19K) - AI orchestration framework to build customizable, production-ready.. Apache-2

GitHub (👨‍💻 280 · 🔀 2K · 📦 850 · 📋 3.7K - 3% open · ⏱️ 13.02.2025):

	git clone https://github.com/deepset-ai/haystack

PyPi (📥 6.1K / month · 📦 5 · ⏱️ 15.12.2021):

	pip install haystack

spark-nlp (🥈35 · ⭐ 3.9K) - State of the Art Natural Language Processing. Apache-2

GitHub (👨‍💻 110 · 🔀 720 · 📦 550 · 📋 910 - 4% open · ⏱️ 30.01.2025):

	git clone https://github.com/JohnSnowLabs/spark-nlp

PyPi (📥 4.4M / month · 📦 37 · ⏱️ 30.01.2025):

	pip install spark-nlp

fastText (🥈34 · ⭐ 26K · 💤) - Library for fast text representation and classification. MIT

GitHub (👨‍💻 68 · 🔀 4.7K · 📦 7.2K · 📋 1.2K - 47% open · ⏱️ 13.03.2024):

	git clone https://github.com/facebookresearch/fastText

PyPi (📥 1.4M / month · 📦 250 · ⏱️ 12.06.2024):

	pip install fasttext

Conda (📥 120K · ⏱️ 19.05.2024):

	conda install -c conda-forge fasttext

stanza (🥈34 · ⭐ 7.4K) - Stanford NLP Python library for tokenization, sentence segmentation,.. Apache-2

GitHub (👨‍💻 69 · 🔀 900 · 📦 3.5K · 📋 920 - 10% open · ⏱️ 24.12.2024):

	git clone https://github.com/stanfordnlp/stanza

PyPi (📥 290K / month · 📦 200 · ⏱️ 24.12.2024):

	pip install stanza

Conda (📥 8.4K · ⏱️ 16.06.2023):

	conda install -c stanfordnlp stanza

qdrant (🥈33 · ⭐ 22K) - Qdrant - High-performance, massive-scale Vector Database and Vector.. Apache-2

GitHub (👨‍💻 130 · 🔀 1.5K · 📥 320K · 📦 120 · 📋 1.4K - 24% open · ⏱️ 11.02.2025):

	git clone https://github.com/qdrant/qdrant

OpenNMT (🥈33 · ⭐ 6.8K · 💤) - Open Source Neural Machine Translation and (Large) Language.. MIT

GitHub (👨‍💻 190 · 🔀 2.2K · 📦 310 · 📋 1.5K - 1% open · ⏱️ 27.06.2024):
```
	git clone https://github.com/OpenNMT/OpenNMT-py
```
PyPi (📥 17K / month · 📦 23 · ⏱️ 18.03.2024):
```
	pip install OpenNMT-py
```

rubrix (🥈33 · ⭐ 4.3K) - Argilla is a collaboration tool for AI engineers and domain experts.. Apache-2

GitHub (👨‍💻 110 · 🔀 400 · 📦 2.9K · 📋 2.2K - 2% open · ⏱️ 06.02.2025):

	git clone https://github.com/recognai/rubrix

PyPi (📥 6.1K / month · ⏱️ 24.10.2022):

	pip install rubrix

Conda (📥 42K · ⏱️ 16.06.2023):

	conda install -c conda-forge rubrix

torchtext (🥈33 · ⭐ 3.5K) - Models, data loaders and abstractions for language processing,.. BSD-3

GitHub (👨‍💻 160 · 🔀 810 · 📋 850 - 39% open · ⏱️ 14.08.2024):

	git clone https://github.com/pytorch/text

PyPi (📥 640K / month · 📦 280 · ⏱️ 24.04.2024):

	pip install torchtext

jellyfish (🥈33 · ⭐ 2.1K) - a python library for doing approximate and phonetic matching of strings. MIT

GitHub (👨‍💻 34 · 🔀 160 · 📦 13K · 📋 140 - 4% open · ⏱️ 31.12.2024):

	git clone https://github.com/jamesturk/jellyfish

PyPi (📥 6.4M / month · 📦 280 · ⏱️ 14.12.2024):

	pip install jellyfish

Conda (📥 1.2M · ⏱️ 17.12.2024):

	conda install -c conda-forge jellyfish

ftfy (🥈32 · ⭐ 3.9K) - Fixes mojibake and other glitches in Unicode text, after the fact. Apache-2

GitHub (👨‍💻 20 · 🔀 120 · 📥 35 · 📦 27K · 📋 150 - 6% open · ⏱️ 30.10.2024):

	git clone https://github.com/rspeer/python-ftfy

PyPi (📥 5.7M / month · 📦 570 · ⏱️ 26.10.2024):

	pip install ftfy

Conda (📥 320K · ⏱️ 23.12.2024):

	conda install -c conda-forge ftfy

snowballstemmer (🥈31 · ⭐ 770) - Snowball compiler and stemming algorithms. BSD-3

GitHub (👨‍💻 35 · 🔀 170 · 📦 10 · 📋 91 - 27% open · ⏱️ 30.01.2025):

	git clone https://github.com/snowballstem/snowball

PyPi (📥 17M / month · 📦 450 · ⏱️ 16.11.2021):

	pip install snowballstemmer

Conda (📥 9.2M · ⏱️ 16.06.2023):

	conda install -c conda-forge snowballstemmer

DeepPavlov (🥈29 · ⭐ 6.8K) - An open source library for deep learning end-to-end dialog.. Apache-2

GitHub (👨‍💻 77 · 🔀 1.2K · 📦 420 · 📋 640 - 4% open · ⏱️ 26.11.2024):

	git clone https://github.com/deepmipt/DeepPavlov

PyPi (📥 12K / month · 📦 4 · ⏱️ 12.08.2024):

	pip install deeppavlov

Dedupe (🥈29 · ⭐ 4.2K) - A python library for accurate and scalable fuzzy matching, record.. MIT

GitHub (👨‍💻 72 · 🔀 550 · 📦 350 · 📋 820 - 9% open · ⏱️ 01.11.2024):

	git clone https://github.com/dedupeio/dedupe

PyPi (📥 88K / month · 📦 19 · ⏱️ 15.08.2024):

	pip install dedupe

Conda (📥 96K · ⏱️ 16.06.2023):

	conda install -c conda-forge dedupe

Sumy (🥈29 · ⭐ 3.5K · 💤) - Module for automatic summarization of text documents and HTML pages. Apache-2

GitHub (👨‍💻 32 · 🔀 530 · 📦 3.5K · 📋 120 - 18% open · ⏱️ 16.05.2024):

	git clone https://github.com/miso-belica/sumy

PyPi (📥 140K / month · 📦 31 · ⏱️ 23.10.2022):

	pip install sumy

Conda (📥 11K · ⏱️ 03.01.2025):

	conda install -c conda-forge sumy

TextDistance (🥈29 · ⭐ 3.4K) - Compute distance between sequences. 30+ algorithms, pure python.. MIT

GitHub (👨‍💻 17 · 🔀 250 · 📥 1.1K · 📦 7.9K · ⏱️ 09.09.2024):

	git clone https://github.com/life4/textdistance

PyPi (📥 920K / month · 📦 99 · ⏱️ 16.07.2024):

	pip install textdistance

Conda (📥 730K · ⏱️ 05.01.2025):

	conda install -c conda-forge textdistance

spacy-transformers (🥈29 · ⭐ 1.4K) - Use pretrained transformers like BERT, XLNet and GPT-2.. MIT spacy

GitHub (👨‍💻 23 · 🔀 170 · 📥 110 · 📦 2.1K · ⏱️ 06.02.2025):

	git clone https://github.com/explosion/spacy-transformers

PyPi (📥 220K / month · 📦 98 · ⏱️ 06.02.2025):

	pip install spacy-transformers

Conda (📥 100K · ⏱️ 09.02.2025):

	conda install -c conda-forge spacy-transformers

Opik (🥈28 · ⭐ 5K) - Debug, evaluate, and monitor your LLM applications, RAG systems, and.. Apache-2

GitHub (👨‍💻 35 · 🔀 330 · 📥 11 · 📋 160 - 24% open · ⏱️ 13.02.2025):

	git clone https://github.com/comet-ml/opik

PyPi (📥 34K / month · 📦 4 · ⏱️ 11.02.2025):

	pip install opik

SciSpacy (🥈28 · ⭐ 1.8K) - A full spaCy pipeline and models for scientific/biomedical documents. Apache-2

GitHub (👨‍💻 37 · 🔀 230 · 📦 1K · 📋 320 - 9% open · ⏱️ 23.11.2024):

	git clone https://github.com/allenai/scispacy

PyPi (📥 33K / month · 📦 34 · ⏱️ 27.10.2024):

	pip install scispacy

CLTK (🥈28 · ⭐ 850) - The Classical Language Toolkit. MIT

GitHub (👨‍💻 120 · 🔀 330 · 📥 110 · 📦 290 · 📋 580 - 6% open · ⏱️ 01.12.2024):

	git clone https://github.com/cltk/cltk

PyPi (📥 5.3K / month · 📦 17 · ⏱️ 01.12.2024):

	pip install cltk

english-words (🥉27 · ⭐ 11K) - A text file containing 479k English words for all your.. Unlicense

GitHub (👨‍💻 34 · 🔀 1.9K · 📦 2 · 📋 150 - 73% open · ⏱️ 06.01.2025):

	git clone https://github.com/dwyl/english-words

PyPi (📥 33K / month · 📦 14 · ⏱️ 24.05.2023):

	pip install english-words

DeepKE (🥉27 · ⭐ 3.7K) - [EMNLP 2022] An Open Toolkit for Knowledge Graph Extraction and.. MIT

GitHub (👨‍💻 33 · 🔀 690 · 📦 24 · 📋 600 - 1% open · ⏱️ 13.02.2025):

	git clone https://github.com/zjunlp/deepke

PyPi (📥 1.5K / month · ⏱️ 21.09.2023):

	pip install deepke

scattertext (🥉26 · ⭐ 2.3K) - Beautiful visualizations of how language differs among document.. Apache-2

GitHub (👨‍💻 14 · 🔀 290 · 📦 650 · 📋 100 - 21% open · ⏱️ 23.09.2024):

	git clone https://github.com/JasonKessler/scattertext

PyPi (📥 14K / month · 📦 5 · ⏱️ 23.09.2024):

	pip install scattertext

Conda (📥 110K · ⏱️ 16.06.2023):

	conda install -c conda-forge scattertext

PyTextRank (🥉26 · ⭐ 2.2K · 💤) - Python implementation of TextRank algorithms (textgraphs) for.. MIT

GitHub (👨‍💻 19 · 🔀 340 · 📦 800 · 📋 100 - 12% open · ⏱️ 21.05.2024):

	git clone https://github.com/DerwenAI/pytextrank

PyPi (📥 68K / month · 📦 19 · ⏱️ 21.02.2024):

	pip install pytextrank

detoxify (🥉24 · ⭐ 1K) - Trained models & code to predict toxic comments on all 3 Jigsaw Toxic.. Apache-2

GitHub (👨‍💻 14 · 🔀 120 · 📥 870K · 📦 780 · 📋 66 - 56% open · ⏱️ 21.01.2025):

	git clone https://github.com/unitaryai/detoxify

PyPi (📥 27K / month · 📦 30 · ⏱️ 01.02.2024):

	pip install detoxify

T5 (🥉23 · ⭐ 6.3K · 💤) - Code for the paper Exploring the Limits of Transfer Learning with.. Apache-2

GitHub (👨‍💻 59 · 🔀 760 · 📋 450 - 23% open · ⏱️ 28.06.2024):

	git clone https://github.com/google-research/text-to-text-transfer-transformer

PyPi (📥 41K / month · 📦 2 · ⏱️ 18.10.2021):

	pip install t5

Sockeye (🥉22 · ⭐ 1.2K) - Sequence-to-sequence framework with a focus on Neural Machine.. Apache-2

GitHub (👨‍💻 60 · 🔀 320 · 📥 21 · 📋 310 - 3% open · ⏱️ 24.10.2024):

	git clone https://github.com/awslabs/sockeye

PyPi (📥 2.3K / month · ⏱️ 03.03.2023):

	pip install sockeye

finetune (🥉22 · ⭐ 710) - Scikit-learn style model finetuning for NLP. MPL-2.0

GitHub (👨‍💻 23 · 🔀 79 · 📦 14 · 📋 140 - 15% open · ⏱️ 10.02.2025):

	git clone https://github.com/IndicoDataSolutions/finetune

PyPi (📥 800 / month · 📦 2 · ⏱️ 29.09.2023):

	pip install finetune

small-text (🥉22 · ⭐ 600) - Active Learning for Text Classification in Python. MIT

GitHub (👨‍💻 9 · 🔀 67 · 📦 33 · 📋 65 - 26% open · ⏱️ 21.01.2025):

	git clone https://github.com/webis-de/small-text

PyPi (📥 1K / month · ⏱️ 24.11.2024):

	pip install small-text

Conda (📥 13K · ⏱️ 05.01.2025):

	conda install -c conda-forge small-text

happy-transformer (🥉22 · ⭐ 530 · 💤) - Happy Transformer makes it easy to fine-tune and.. Apache-2 huggingface

GitHub (👨‍💻 14 · 🔀 68 · 📦 300 · 📋 130 - 16% open · ⏱️ 19.03.2024):

	git clone https://github.com/EricFillion/happy-transformer

PyPi (📥 2.8K / month · 📦 5 · ⏱️ 05.08.2023):

	pip install happytransformer

fast-bert (🥉21 · ⭐ 1.9K) - Super easy library for BERT based NLP models. Apache-2

GitHub (👨‍💻 37 · 🔀 340 · 📋 260 - 63% open · ⏱️ 19.08.2024):

	git clone https://github.com/utterworks/fast-bert

PyPi (📥 2.6K / month · ⏱️ 19.08.2024):

	pip install fast-bert

UForm (🥉20 · ⭐ 1.1K) - Pocket-Sized Multimodal AI for content understanding and.. Apache-2

GitHub (👨‍💻 19 · 🔀 63 · 📥 540 · 📦 6 · 📋 32 - 31% open · ⏱️ 03.01.2025):

	git clone https://github.com/unum-cloud/uform

PyPi (📥 1.3K / month · 📦 2 · ⏱️ 03.01.2025):

	pip install uform

VizSeq (🥉16 · ⭐ 440) - An Analysis Toolkit for Natural Language Generation (Translation,.. MIT

GitHub (👨‍💻 4 · 🔀 61 · 📦 12 · 📋 16 - 43% open · ⏱️ 28.09.2024):

	git clone https://github.com/facebookresearch/vizseq

PyPi (📥 250 / month · ⏱️ 07.08.2020):

	pip install vizseq

Show 56 hidden projects...

AllenNLP (🥈36 · ⭐ 12K · 💀) - An open-source NLP research library, built on PyTorch. Apache-2

ParlAI (🥈32 · ⭐ 10K · 💀) - A framework for training and evaluating AI models on a variety of.. MIT

fuzzywuzzy (🥈32 · ⭐ 9.2K · 💀) - Fuzzy String Matching in Python. ❗️GPL-2.0

nlpaug (🥈30 · ⭐ 4.5K · 💀) - Data augmentation for NLP. MIT

GluonNLP (🥈29 · ⭐ 2.6K · 💀) - Toolkit that enables easy text preprocessing, datasets.. Apache-2

langid (🥈29 · ⭐ 2.4K · 💀) - Stand-alone language identification system. BSD-3

vaderSentiment (🥈28 · ⭐ 4.6K · 💀) - VADER Sentiment Analysis. VADER (Valence Aware Dictionary.. MIT

fastNLP (🥈28 · ⭐ 3.1K · 💀) - fastNLP: A Modularized and Extensible NLP Framework. Currently.. Apache-2

Ciphey (🥉27 · ⭐ 19K · 💀) - Automatically decrypt encryptions without knowing the key or cipher,.. MIT

textacy (🥉27 · ⭐ 2.2K · 💀) - NLP, before and after spaCy. ❗Unlicensed

FARM (🥉27 · ⭐ 1.8K · 💀) - Fast & easy transfer learning for NLP. Harvesting language.. Apache-2

pySBD (🥉27 · ⭐ 830 · 💀) - pySBD (Python Sentence Boundary Disambiguation) is a rule-based sentence.. MIT

flashtext (🥉26 · ⭐ 5.6K · 💀) - Extract Keywords from sentence or Replace keywords in sentences. MIT

polyglot (🥉26 · ⭐ 2.3K · 💀) - Multilingual text (NLP) processing toolkit. ❗️GPL-3.0

underthesea (🥉26 · ⭐ 1.5K) - Underthesea - Vietnamese NLP Toolkit. ❗️GPL-3.0

PyText (🥉25 · ⭐ 6.3K · 💀) - A natural language modeling framework based on PyTorch. BSD-3

OpenPrompt (🥉25 · ⭐ 4.5K · 💀) - An Open-Source Framework for Prompt-Learning. Apache-2

Snips NLU (🥉25 · ⭐ 3.9K · 💀) - Snips Python library to extract meaning from text. Apache-2

neuralcoref (🥉25 · ⭐ 2.9K · 💀) - Fast Coreference Resolution in spaCy with Neural Networks. MIT

pytorch-nlp (🥉25 · ⭐ 2.2K · 💀) - Basic Utilities for PyTorch Natural Language Processing.. BSD-3

textgenrnn (🥉24 · ⭐ 4.9K · 💀) - Easily train your own text-generating neural network of any.. MIT

MatchZoo (🥉24 · ⭐ 3.8K · 💀) - Facilitating the design, comparison and sharing of deep.. Apache-2

promptsource (🥉24 · ⭐ 2.8K · 💀) - Toolkit for creating, sharing and using natural language.. Apache-2

Kashgari (🥉24 · ⭐ 2.4K · 💀) - Kashgari is a production-level NLP Transfer learning.. Apache-2

sense2vec (🥉24 · ⭐ 1.6K · 💀) - Contextually-keyed word vectors. MIT

jiant (🥉23 · ⭐ 1.7K · 💀) - jiant is an nlp toolkit. MIT

whoosh (🥉23 · ⭐ 600 · 💀) - Pure-Python full-text search library. ❗️BSD-1-Clause

gpt-2-simple (🥉22 · ⭐ 3.4K · 💀) - Python package to easily retrain OpenAIs GPT-2 text-.. MIT

NLP Architect (🥉22 · ⭐ 2.9K · 💀) - A model library for exploring state-of-the-art deep.. Apache-2

Texthero (🥉22 · ⭐ 2.9K · 💀) - Text preprocessing, representation and visualization from zero to.. MIT

Texar (🥉22 · ⭐ 2.4K · 💀) - Toolkit for Machine Learning, Natural Language Processing, and.. Apache-2

YouTokenToMe (🥉22 · ⭐ 960 · 💀) - Unsupervised text tokenizer focused on computational efficiency. MIT

stop-words (🥉22 · ⭐ 160 · 💀) - Get list of common stop words in various languages in Python. BSD-3

anaGo (🥉21 · ⭐ 1.5K · 💀) - Bidirectional LSTM-CRF and ELMo for Named-Entity Recognition,.. MIT

DeepMatcher (🥉20 · ⭐ 5.2K · 💀) - Python package for performing Entity and Text Matching using.. BSD-3

lightseq (🥉20 · ⭐ 3.2K · 💀) - LightSeq: A High Performance Library for Sequence Processing.. Apache-2

DELTA (🥉20 · ⭐ 1.6K · 💀) - DELTA is a deep learning based natural language and speech.. Apache-2

textpipe (🥉20 · ⭐ 300 · 💀) - Textpipe: clean and extract metadata from text. MIT

pyfasttext (🥉20 · ⭐ 230 · 💀) - Yet another Python binding for fastText. ❗️GPL-3.0

numerizer (🥉20 · ⭐ 220) - A Python module to convert natural language numerics into ints and.. MIT

Camphr (🥉19 · ⭐ 340 · 💀) - Camphr - NLP libary for creating pipeline components. Apache-2 spacy

NeuroNER (🥉18 · ⭐ 1.7K · 💀) - Named-entity recognition using neural networks. Easy-to-use and.. MIT

nboost (🥉18 · ⭐ 680 · 💀) - NBoost is a scalable, search-api-boosting platform for deploying.. Apache-2

fastT5 (🥉18 · ⭐ 570 · 💀) - boost inference speed of T5 models by 5x & reduce the model size.. Apache-2

textaugment (🥉18 · ⭐ 410 · 💀) - TextAugment: Text Augmentation Library. MIT

skift (🥉17 · ⭐ 240 · 💀) - scikit-learn wrappers for Python fastText. MIT

OpenNRE (🥉16 · ⭐ 4.4K · 💀) - An Open-Source Package for Neural Relation Extraction (NRE). MIT

BLINK (🥉15 · ⭐ 1.2K · 💀) - Entity Linker solution. MIT

TextBox (🥉15 · ⭐ 1.1K · 💀) - TextBox 2.0 is a text generation library with pre-trained language.. MIT

Translate (🥉15 · ⭐ 830 · 💀) - Translate - a PyTorch Language Library. BSD-3

ONNX-T5 (🥉15 · ⭐ 250 · 💀) - Summarization, translation, sentiment-analysis, text-generation.. Apache-2

NeuralQA (🥉15 · ⭐ 230 · 💀) - NeuralQA: A Usable Library for Question Answering on Large Datasets.. MIT

Headliner (🥉15 · ⭐ 230 · 💀) - Easy training and deployment of seq2seq models. MIT

TransferNLP (🥉14 · ⭐ 290 · 💀) - NLP library designed for reproducible experimentation.. MIT

textvec (🥉13 · ⭐ 190 · 💀) - Text vectorization tool to outperform TFIDF for classification.. MIT

spacy-dbpedia-spotlight (🥉13 · ⭐ 110 · 💀) - A spaCy wrapper for DBpedia Spotlight. MIT spacy

Image Data

Libraries for image & video processing, manipulation, and augmentation as well as libraries for computer vision tasks such as facial recognition, object detection, and classification.

Pillow (🥇48 · ⭐ 13K) - Python Imaging Library (Fork). ❗️PIL

GitHub (👨‍💻 480 · 🔀 2.3K · 📦 2.1M · 📋 3.3K - 3% open · ⏱️ 09.02.2025):
```
	git clone https://github.com/python-pillow/Pillow
```
PyPi (📥 130M / month · 📦 11K · ⏱️ 02.01.2025):
```
	pip install Pillow
```
Conda (📥 50M · ⏱️ 03.01.2025):
```
	conda install -c conda-forge pillow
```

MoviePy (🥇43 · ⭐ 13K) - Video editing with Python. MIT

GitHub (👨‍💻 180 · 🔀 1.6K · 📦 54K · 📋 2K - 23% open · ⏱️ 06.02.2025):

	git clone https://github.com/Zulko/moviepy

PyPi (📥 2.7M / month · 📦 1K · ⏱️ 10.01.2025):

	pip install moviepy

Conda (📥 290K · ⏱️ 16.06.2023):

	conda install -c conda-forge moviepy

PyTorch Image Models (🥇42 · ⭐ 33K) - The largest collection of PyTorch image encoders /.. Apache-2

GitHub (👨‍💻 170 · 🔀 4.8K · 📥 7.5M · 📦 47K · 📋 960 - 5% open · ⏱️ 31.01.2025):

	git clone https://github.com/rwightman/pytorch-image-models

PyPi (📥 6.4M / month · 📦 1.1K · ⏱️ 19.01.2025):

	pip install timm

Conda (📥 320K · ⏱️ 20.01.2025):

	conda install -c conda-forge timm

torchvision (🥇41 · ⭐ 17K) - Datasets, Transforms and Models specific to Computer Vision. BSD-3

GitHub (👨‍💻 620 · 🔀 7K · 📥 40K · 📦 21 · 📋 3.6K - 30% open · ⏱️ 07.02.2025):

	git clone https://github.com/pytorch/vision

PyPi (📥 15M / month · 📦 6.3K · ⏱️ 29.01.2025):

	pip install torchvision

Conda (📥 2.3M · ⏱️ 20.01.2025):

	conda install -c conda-forge torchvision

Albumentations (🥇41 · ⭐ 15K) - Fast and flexible image augmentation library. Paper about.. MIT

GitHub (👨‍💻 160 · 🔀 1.7K · 📦 32K · 📋 1.1K - 13% open · ⏱️ 11.02.2025):

	git clone https://github.com/albumentations-team/albumentations

PyPi (📥 6M / month · 📦 650 · ⏱️ 11.02.2025):

	pip install albumentations

Conda (📥 230K · ⏱️ 12.02.2025):

	conda install -c conda-forge albumentations

deepface (🥇38 · ⭐ 18K) - A Lightweight Face Recognition and Facial Attribute Analysis (Age,.. MIT

GitHub (👨‍💻 78 · 🔀 2.4K · 📦 5.5K · 📋 1.2K - 0% open · ⏱️ 04.02.2025):

	git clone https://github.com/serengil/deepface

PyPi (📥 420K / month · 📦 44 · ⏱️ 17.08.2024):

	pip install deepface

imageio (🥇38 · ⭐ 1.5K) - Python library for reading and writing image data. BSD-2

GitHub (👨‍💻 120 · 🔀 300 · 📥 1.5K · 📦 160K · 📋 610 - 16% open · ⏱️ 20.01.2025):

	git clone https://github.com/imageio/imageio

PyPi (📥 24M / month · 📦 2.6K · ⏱️ 20.01.2025):

	pip install imageio

Conda (📥 7.5M · ⏱️ 30.01.2025):

	conda install -c conda-forge imageio

Kornia (🥈37 · ⭐ 10K) - Geometric Computer Vision Library for Spatial AI. Apache-2

GitHub (👨‍💻 280 · 🔀 980 · 📥 1.7K · 📦 14K · 📋 960 - 30% open · ⏱️ 13.02.2025):

	git clone https://github.com/kornia/kornia

PyPi (📥 1.9M / month · 📦 310 · ⏱️ 11.01.2025):

	pip install kornia

Conda (📥 170K · ⏱️ 11.01.2025):

	conda install -c conda-forge kornia

opencv-python (🥈37 · ⭐ 4.7K) - Automated CI toolchain to produce precompiled opencv-python,.. MIT

GitHub (👨‍💻 53 · 🔀 870 · 📦 510K · 📋 840 - 16% open · ⏱️ 16.01.2025):

	git clone https://github.com/opencv/opencv-python

PyPi (📥 16M / month · 📦 12K · ⏱️ 16.01.2025):

	pip install opencv-python

MMDetection (🥈36 · ⭐ 30K · 💤) - OpenMMLab Detection Toolbox and Benchmark. Apache-2

GitHub (👨‍💻 480 · 🔀 9.5K · 📦 3.5K · 📋 8.6K - 21% open · ⏱️ 05.02.2024):

	git clone https://github.com/open-mmlab/mmdetection

PyPi (📥 180K / month · 📦 82 · ⏱️ 05.01.2024):

	pip install mmdet

InsightFace (🥈36 · ⭐ 24K) - State-of-the-art 2D and 3D Face Analysis Project. MIT

GitHub (👨‍💻 63 · 🔀 5.4K · 📥 6.5M · 📦 3.4K · 📋 2.5K - 45% open · ⏱️ 13.02.2025):

	git clone https://github.com/deepinsight/insightface

PyPi (📥 220K / month · 📦 30 · ⏱️ 17.12.2022):

	pip install insightface

detectron2 (🥈34 · ⭐ 31K) - Detectron2 is a platform for object detection, segmentation.. Apache-2

GitHub (👨‍💻 280 · 🔀 7.5K · 📦 2.3K · 📋 3.6K - 14% open · ⏱️ 14.01.2025):

	git clone https://github.com/facebookresearch/detectron2

PyPi (📦 13 · ⏱️ 06.02.2020):

	pip install detectron2

Conda (📥 600K · ⏱️ 06.11.2024):

	conda install -c conda-forge detectron2

Wand (🥈34 · ⭐ 1.4K) - The ctypes-based simple ImageMagick binding for Python. MIT

GitHub (👨‍💻 110 · 🔀 200 · 📥 52K · 📦 20K · 📋 430 - 6% open · ⏱️ 02.02.2025):

	git clone https://github.com/emcconville/wand

PyPi (📥 1.1M / month · 📦 260 · ⏱️ 03.11.2023):

	pip install wand

Conda (📥 100K · ⏱️ 16.06.2023):

	conda install -c conda-forge wand

PaddleSeg (🥈33 · ⭐ 8.8K) - Easy-to-use image segmentation library with awesome pre-.. Apache-2

GitHub (👨‍💻 130 · 🔀 1.7K · 📦 1.4K · 📋 2.2K - 0% open · ⏱️ 25.12.2024):

	git clone https://github.com/PaddlePaddle/PaddleSeg

PyPi (📥 2K / month · 📦 7 · ⏱️ 30.11.2022):

	pip install paddleseg

ImageHash (🥈32 · ⭐ 3.5K) - A Python Perceptual Image Hashing Module. BSD-2

GitHub (👨‍💻 27 · 🔀 340 · 📦 16K · 📋 150 - 15% open · ⏱️ 09.10.2024):

	git clone https://github.com/JohannesBuchner/imagehash

PyPi (📥 1.5M / month · 📦 270 · ⏱️ 01.02.2025):

	pip install ImageHash

Conda (📥 420K · ⏱️ 03.02.2025):

	conda install -c conda-forge imagehash

lightly (🥈32 · ⭐ 3.3K) - A python library for self-supervised learning on images. MIT

GitHub (👨‍💻 64 · 🔀 290 · 📦 390 · 📋 600 - 13% open · ⏱️ 13.02.2025):

	git clone https://github.com/lightly-ai/lightly

PyPi (📥 41K / month · 📦 18 · ⏱️ 28.01.2025):

	pip install lightly

vit-pytorch (🥈31 · ⭐ 22K · 📉) - Implementation of Vision Transformer, a simple way to.. MIT

GitHub (👨‍💻 23 · 🔀 3.2K · 📦 600 · 📋 280 - 49% open · ⏱️ 19.01.2025):

	git clone https://github.com/lucidrains/vit-pytorch

PyPi (📥 26K / month · 📦 17 · ⏱️ 19.01.2025):

	pip install vit-pytorch

imageai (🥈30 · ⭐ 8.7K · 💤) - A python library built to empower developers to build applications.. MIT

GitHub (👨‍💻 19 · 🔀 2.2K · 📥 950K · 📦 1.7K · 📋 760 - 41% open · ⏱️ 20.02.2024):

	git clone https://github.com/OlafenwaMoses/ImageAI

PyPi (📥 10K / month · 📦 19 · ⏱️ 02.01.2023):

	pip install imageai

Conda (📥 8.7K · ⏱️ 16.06.2023):

	conda install -c conda-forge imageai

CellProfiler (🥈30 · ⭐ 950) - An open-source application for biological image analysis. BSD-3

GitHub (👨‍💻 140 · 🔀 390 · 📥 8.3K · 📦 27 · 📋 3.3K - 9% open · ⏱️ 10.02.2025):

	git clone https://github.com/CellProfiler/CellProfiler

PyPi (📥 1.5K / month · 📦 2 · ⏱️ 16.09.2024):

	pip install cellprofiler

sahi (🥈29 · ⭐ 4.3K) - Framework agnostic sliced/tiled inference + interactive ui + error analysis.. MIT

GitHub (👨‍💻 48 · 🔀 600 · 📥 32K · 📦 1.6K · ⏱️ 18.12.2024):

	git clone https://github.com/obss/sahi

PyPi (📥 170K / month · 📦 31 · ⏱️ 16.12.2024):

	pip install sahi

Conda (📥 89K · ⏱️ 18.12.2024):

	conda install -c conda-forge sahi

doctr (🥈29 · ⭐ 4.3K) - docTR (Document Text Recognition) - a seamless, high-.. Apache-2

GitHub (👨‍💻 56 · 🔀 470 · 📥 4.9M · 📋 390 - 6% open · ⏱️ 03.02.2025):

	git clone https://github.com/mindee/doctr

PyPi (📥 64K / month · 📦 14 · ⏱️ 30.01.2025):

	pip install python-doctr

PaddleDetection (🥉28 · ⭐ 13K · 📉) - Object Detection toolkit based on PaddlePaddle. It.. Apache-2

GitHub (👨‍💻 180 · 🔀 2.9K · 📋 5.5K - 22% open · ⏱️ 03.12.2024):

	git clone https://github.com/PaddlePaddle/PaddleDetection

PyPi (📥 950 / month · 📦 2 · ⏱️ 19.09.2022):

	pip install paddledet

Face Alignment (🥉28 · ⭐ 7.2K) - 2D and 3D Face alignment library build using pytorch. BSD-3

GitHub (👨‍💻 26 · 🔀 1.4K · 📦 21 · 📋 320 - 24% open · ⏱️ 30.08.2024):

	git clone https://github.com/1adrianb/face-alignment

PyPi (📥 74K / month · 📦 10 · ⏱️ 17.08.2023):

	pip install face-alignment

mtcnn (🥉28 · ⭐ 2.3K) - MTCNN face detection implementation for TensorFlow, as a PIP package. MIT

GitHub (👨‍💻 16 · 🔀 530 · 📥 30 · 📦 7.3K · 📋 130 - 37% open · ⏱️ 08.10.2024):

	git clone https://github.com/ipazc/mtcnn

PyPi (📥 140K / month · 📦 73 · ⏱️ 08.10.2024):

	pip install mtcnn

Conda (📥 14K · ⏱️ 16.06.2023):

	conda install -c conda-forge mtcnn

facenet-pytorch (🥉27 · ⭐ 4.7K) - Pretrained Pytorch face detection (MTCNN) and facial.. MIT

GitHub (👨‍💻 18 · 🔀 940 · 📥 1.6M · 📦 2.8K · 📋 190 - 41% open · ⏱️ 02.08.2024):

	git clone https://github.com/timesler/facenet-pytorch

PyPi (📥 97K / month · 📦 51 · ⏱️ 29.04.2024):

	pip install facenet-pytorch

vidgear (🥉27 · ⭐ 3.4K · 💤) - A High-performance cross-platform Video Processing Python.. Apache-2

GitHub (👨‍💻 14 · 🔀 260 · 📥 2.2K · 📦 680 · 📋 300 - 2% open · ⏱️ 22.06.2024):

	git clone https://github.com/abhiTronix/vidgear

PyPi (📥 20K / month · 📦 15 · ⏱️ 22.06.2024):

	pip install vidgear

mahotas (🥉27 · ⭐ 860 · 💤) - Computer Vision in Python. MIT

GitHub (👨‍💻 35 · 🔀 150 · 📦 1.4K · 📋 91 - 23% open · ⏱️ 17.07.2024):

	git clone https://github.com/luispedro/mahotas

PyPi (📥 20K / month · 📦 63 · ⏱️ 17.07.2024):

	pip install mahotas

Conda (📥 590K · ⏱️ 18.07.2024):

	conda install -c conda-forge mahotas

Norfair (🥉26 · ⭐ 2.4K · 💤) - Lightweight Python library for adding real-time multi-object.. BSD-3

GitHub (👨‍💻 31 · 🔀 250 · 📥 340 · 📦 270 · 📋 170 - 14% open · ⏱️ 27.07.2024):

	git clone https://github.com/tryolabs/norfair

PyPi (📥 23K / month · 📦 9 · ⏱️ 30.05.2022):

	pip install norfair

Image Deduplicator (🥉25 · ⭐ 5.2K) - Finding duplicate images made easy!. Apache-2

GitHub (👨‍💻 16 · 🔀 460 · 📦 170 · 📋 130 - 31% open · ⏱️ 19.12.2024):

	git clone https://github.com/idealo/imagededup

PyPi (📥 17K / month · 📦 5 · ⏱️ 28.04.2023):

	pip install imagededup

pytorchvideo (🥉25 · ⭐ 3.4K) - A deep learning library for video understanding research. Apache-2

GitHub (👨‍💻 58 · 🔀 410 · 📋 210 - 50% open · ⏱️ 25.01.2025):

	git clone https://github.com/facebookresearch/pytorchvideo

PyPi (📥 42K / month · 📦 24 · ⏱️ 20.01.2022):

	pip install pytorchvideo

tensorflow-graphics (🥉25 · ⭐ 2.8K) - TensorFlow Graphics: Differentiable Graphics Layers.. Apache-2

GitHub (👨‍💻 39 · 🔀 370 · 📋 240 - 60% open · ⏱️ 03.02.2025):

	git clone https://github.com/tensorflow/graphics

PyPi (📥 91K / month · 📦 11 · ⏱️ 03.12.2021):

	pip install tensorflow-graphics

kubric (🥉25 · ⭐ 2.4K) - A data generation pipeline for creating semi-realistic synthetic.. Apache-2

GitHub (👨‍💻 31 · 🔀 230 · 📦 7 · 📋 190 - 33% open · ⏱️ 29.11.2024):

	git clone https://github.com/google-research/kubric

PyPi (📥 20K / month · ⏱️ 27.12.2023):

	pip install kubric-nightly

pyvips (🥉25 · ⭐ 680) - python binding for libvips using cffi. MIT

GitHub (👨‍💻 16 · 🔀 50 · 📦 930 · 📋 450 - 42% open · ⏱️ 29.10.2024):

	git clone https://github.com/libvips/pyvips

PyPi (📥 69K / month · 📦 77 · ⏱️ 28.04.2024):

	pip install pyvips

Conda (📥 200K · ⏱️ 06.09.2024):

	conda install -c conda-forge pyvips

MMF (🥉24 · ⭐ 5.5K) - A modular framework for vision & language multimodal research from.. BSD-3

GitHub (👨‍💻 120 · 🔀 920 · 📦 21 · 📋 690 - 21% open · ⏱️ 03.02.2025):

	git clone https://github.com/facebookresearch/mmf

PyPi (📥 730 / month · 📦 1 · ⏱️ 12.06.2020):

	pip install mmf

segmentation_models (🥉24 · ⭐ 4.8K) - Segmentation models with pretrained backbones. Keras.. MIT

GitHub (👨‍💻 15 · 🔀 1K · 📋 540 - 50% open · ⏱️ 21.08.2024):

	git clone https://github.com/qubvel/segmentation_models

PyPi (📥 35K / month · 📦 28 · ⏱️ 10.01.2020):

	pip install segmentation_models

vissl (🥉23 · ⭐ 3.3K · 💤) - VISSL is FAIRs library of extensible, modular and scalable.. MIT

GitHub (👨‍💻 38 · 🔀 330 · 📦 58 · 📋 190 - 43% open · ⏱️ 03.03.2024):

	git clone https://github.com/facebookresearch/vissl

PyPi (📥 180 / month · 📦 1 · ⏱️ 02.11.2021):

	pip install vissl

ffcv (🥉23 · ⭐ 2.9K · 💤) - FFCV: Fast Forward Computer Vision (and other ML workloads!). Apache-2

GitHub (👨‍💻 31 · 🔀 180 · 📦 60 · 📋 290 - 38% open · ⏱️ 06.05.2024):

	git clone https://github.com/libffcv/ffcv

PyPi (📥 940 / month · 📦 1 · ⏱️ 28.01.2022):

	pip install ffcv

icevision (🥉22 · ⭐ 860) - An Agnostic Computer Vision Framework - Pluggable to any Training.. Apache-2

GitHub (👨‍💻 41 · 🔀 130 · 📋 570 - 10% open · ⏱️ 31.10.2024):

	git clone https://github.com/airctic/icevision

PyPi (📥 3.2K / month · 📦 6 · ⏱️ 10.02.2022):

	pip install icevision

DE⫶TR (🥉21 · ⭐ 14K · 💤) - End-to-End Object Detection with Transformers. Apache-2

GitHub (👨‍💻 27 · 🔀 2.4K · 📦 21 · 📋 540 - 47% open · ⏱️ 12.03.2024):

	git clone https://github.com/facebookresearch/detr

Image Super-Resolution (🥉21 · ⭐ 4.7K) - Super-scale your images and run experiments with.. Apache-2

GitHub (👨‍💻 11 · 🔀 750 · 📋 220 - 48% open · ⏱️ 18.12.2024):
```
	git clone https://github.com/idealo/image-super-resolution
```
PyPi (📥 5.6K / month · 📦 5 · ⏱️ 08.01.2020):
```
	pip install ISR
```
Docker Hub (📥 270 · ⭐ 1 · ⏱️ 01.04.2019):
```
	docker pull idealo/image-super-resolution-gpu
```

PySlowFast (🥉20 · ⭐ 6.8K) - PySlowFast: video understanding codebase from FAIR for.. Apache-2

GitHub (👨‍💻 34 · 🔀 1.2K · 📦 22 · 📋 700 - 58% open · ⏱️ 26.11.2024):

	git clone https://github.com/facebookresearch/SlowFast

PyPi (📥 64 / month · ⏱️ 15.01.2020):

	pip install pyslowfast

scenic (🥉18 · ⭐ 3.4K) - Scenic: A Jax Library for Computer Vision Research and Beyond. Apache-2

GitHub (👨‍💻 91 · 🔀 440 · 📋 270 - 55% open · ⏱️ 31.01.2025):

	git clone https://github.com/google-research/scenic

Show 22 hidden projects...

scikit-image (🥇42 · ⭐ 6.2K) - Image processing in Python. ❗Unlicensed
glfw (🥈37 · ⭐ 13K) - A multi-platform library for OpenGL, OpenGL ES, Vulkan, window and input. ❗️Zlib
imgaug (🥈36 · ⭐ 15K · 💀) - Image augmentation for machine learning experiments. MIT
Face Recognition (🥈35 · ⭐ 54K · 💀) - The worlds simplest facial recognition api for Python.. MIT
PyTorch3D (🥈34 · ⭐ 9K) - PyTorch3D is FAIRs library of reusable components for deep.. ❗Unlicensed
imutils (🥈32 · ⭐ 4.6K · 💀) - A series of convenience functions to make basic image processing.. MIT
GluonCV (🥈29 · ⭐ 5.9K · 💀) - Gluon CV Toolkit. Apache-2
Augmentor (🥉27 · ⭐ 5.1K · 💀) - Image augmentation library in Python for machine learning. MIT
layout-parser (🥉27 · ⭐ 5.1K · 💀) - A Unified Toolkit for Deep Learning Based Document Image.. Apache-2
chainercv (🥉27 · ⭐ 1.5K · 💀) - ChainerCV: a Library for Deep Learning in Computer Vision. MIT
Pillow-SIMD (🥉25 · ⭐ 2.2K) - The friendly PIL fork. ❗️PIL
deep-daze (🥉23 · ⭐ 4.4K · 💀) - Simple command line tool for text to image generation using.. MIT
Luminoth (🥉23 · ⭐ 2.4K · 💀) - Deep Learning toolkit for Computer Vision. BSD-3
Classy Vision (🥉22 · ⭐ 1.6K · 💀) - An end-to-end PyTorch framework for image and video.. MIT
image-match (🥉21 · ⭐ 3K · 💀) - Quickly search over billions of images. Apache-2
detecto (🥉21 · ⭐ 620 · 💀) - Build fully-functioning computer vision models with PyTorch. MIT
nude.py (🥉20 · ⭐ 930 · 💀) - Nudity detection with Python. MIT
pycls (🥉17 · ⭐ 2.1K · 💀) - Codebase for Image Classification Research, written in PyTorch. MIT
Caer (🥉17 · ⭐ 780 · 💀) - A lightweight Computer Vision library. Scale your models, not boilerplate. MIT
solt (🥉17 · ⭐ 260) - Streaming over lightweight data transformations. MIT
Torch Points 3D (🥉17 · ⭐ 230 · 💀) - Pytorch framework for doing deep learning on point.. BSD-3
HugsVision (🥉16 · ⭐ 190 · 💀) - HugsVision is a easy to use huggingface wrapper for state-of-.. MIT huggingface

Graph Data

Libraries for graph processing, clustering, embedding, and machine learning tasks.

networkx (🥇45 · ⭐ 15K) - Network Analysis in Python. BSD-3

GitHub (👨‍💻 770 · 🔀 3.3K · 📥 79 · 📦 350K · 📋 3.4K - 10% open · ⏱️ 12.02.2025):
```
	git clone https://github.com/networkx/networkx
```
PyPi (📥 69M / month · 📦 9.6K · ⏱️ 21.10.2024):
```
	pip install networkx
```
Conda (📥 20M · ⏱️ 25.12.2024):
```
	conda install -c conda-forge networkx
```

PyTorch Geometric (🥇40 · ⭐ 22K) - Graph Neural Network Library for PyTorch. MIT

GitHub (👨‍💻 540 · 🔀 3.8K · 📦 7.8K · 📋 3.8K - 29% open · ⏱️ 13.02.2025):

	git clone https://github.com/pyg-team/pytorch_geometric

PyPi (📥 420K / month · 📦 360 · ⏱️ 26.09.2024):

	pip install torch-geometric

Conda (📥 140K · ⏱️ 19.12.2024):

	conda install -c conda-forge pytorch_geometric

dgl (🥇36 · ⭐ 14K) - Python package built to ease deep learning on graph, on top of existing DL.. Apache-2

GitHub (👨‍💻 300 · 🔀 3K · 📦 340 · 📋 2.9K - 18% open · ⏱️ 11.02.2025):

	git clone https://github.com/dmlc/dgl

PyPi (📥 100K / month · 📦 150 · ⏱️ 13.05.2024):

	pip install dgl

pygraphistry (🥈31 · ⭐ 2.2K) - PyGraphistry is a Python library to quickly load, shape,.. BSD-3

GitHub (👨‍💻 45 · 🔀 210 · 📦 140 · 📋 360 - 53% open · ⏱️ 06.02.2025):

	git clone https://github.com/graphistry/pygraphistry

PyPi (📥 18K / month · 📦 6 · ⏱️ 06.02.2025):

	pip install graphistry

PyKEEN (🥈31 · ⭐ 1.7K) - A Python library for learning and evaluating knowledge graph embeddings. MIT

GitHub (👨‍💻 41 · 🔀 190 · 📥 230 · 📦 280 · 📋 580 - 19% open · ⏱️ 11.02.2025):

	git clone https://github.com/pykeen/pykeen

PyPi (📥 15K / month · 📦 19 · ⏱️ 29.10.2024):

	pip install pykeen

ogb (🥈28 · ⭐ 2K) - Benchmark datasets, data loaders, and evaluators for graph machine learning. MIT

GitHub (👨‍💻 32 · 🔀 400 · 📦 2.3K · 📋 300 - 9% open · ⏱️ 09.12.2024):

	git clone https://github.com/snap-stanford/ogb

PyPi (📥 32K / month · 📦 22 · ⏱️ 02.11.2022):

	pip install ogb

Conda (📥 47K · ⏱️ 22.12.2024):

	conda install -c conda-forge ogb

pytorch_geometric_temporal (🥈26 · ⭐ 2.7K · 📈) - PyTorch Geometric Temporal: Spatiotemporal Signal.. MIT

GitHub (👨‍💻 36 · 🔀 370 · 📋 200 - 20% open · ⏱️ 09.02.2025):

	git clone https://github.com/benedekrozemberczki/pytorch_geometric_temporal

PyPi (📥 2.8K / month · 📦 7 · ⏱️ 04.09.2022):

	pip install torch-geometric-temporal

AmpliGraph (🥈26 · ⭐ 2.2K · 💤) - Python library for Representation Learning on Knowledge.. Apache-2

GitHub (👨‍💻 21 · 🔀 250 · 📦 63 · 📋 230 - 13% open · ⏱️ 28.02.2024):

	git clone https://github.com/Accenture/AmpliGraph

PyPi (📥 1.5K / month · 📦 2 · ⏱️ 26.02.2024):

	pip install ampligraph

PyTorch-BigGraph (🥈23 · ⭐ 3.4K · 💤) - Generate embeddings from large-scale graph-structured.. BSD-3

GitHub (👨‍💻 32 · 🔀 450 · 📥 220 · 📋 200 - 32% open · ⏱️ 03.03.2024):

	git clone https://github.com/facebookresearch/PyTorch-BigGraph

PyPi (📥 300K / month · 📦 2 · ⏱️ 14.10.2019):

	pip install torchbiggraph

Node2Vec (🥈23 · ⭐ 1.3K) - Implementation of the node2vec algorithm. MIT

GitHub (👨‍💻 16 · 🔀 250 · 📦 800 · 📋 93 - 5% open · ⏱️ 02.08.2024):

	git clone https://github.com/eliorc/node2vec

PyPi (📥 22K / month · 📦 31 · ⏱️ 02.08.2024):

	pip install node2vec

Conda (📥 33K · ⏱️ 16.06.2023):

	conda install -c conda-forge node2vec

torch-cluster (🥈23 · ⭐ 840) - PyTorch Extension Library of Optimized Graph Cluster.. MIT

GitHub (👨‍💻 38 · 🔀 150 · 📋 180 - 17% open · ⏱️ 10.01.2025):

	git clone https://github.com/rusty1s/pytorch_cluster

PyPi (📥 16K / month · 📦 62 · ⏱️ 12.10.2023):

	pip install torch-cluster

Conda (📥 320K · ⏱️ 12.02.2025):

	conda install -c conda-forge pytorch_cluster

AutoGL (🥉15 · ⭐ 1.1K · 💤) - An autoML framework & toolkit for machine learning on graphs. Apache-2

GitHub (👨‍💻 16 · 🔀 120 · 📋 39 - 35% open · ⏱️ 05.02.2024):

	git clone https://github.com/THUMNLab/AutoGL

PyPi (⏱️ 23.12.2020):

	pip install auto-graph-learning

GraphVite (🥉14 · ⭐ 1.2K · 💤) - GraphVite: A General and High-performance Graph Embedding.. Apache-2

GitHub (👨‍💻 1 · 🔀 150 · 📋 110 - 46% open · ⏱️ 14.06.2024):

	git clone https://github.com/DeepGraphLearning/graphvite

Conda (📥 5K · ⏱️ 16.06.2023):

	conda install -c milagraph graphvite

Show 23 hidden projects...

igraph (🥇33 · ⭐ 1.3K) - Python interface for igraph. ❗️GPL-2.0

Spektral (🥈28 · ⭐ 2.4K · 💀) - Graph Neural Networks with Keras and Tensorflow 2. MIT

StellarGraph (🥈27 · ⭐ 3K · 💀) - StellarGraph - Machine Learning on Graphs. Apache-2

pygal (🥈27 · ⭐ 2.7K) - PYthon svg GrAph plotting Library. ❗️LGPL-3.0

Paddle Graph Learning (🥈26 · ⭐ 1.6K · 💀) - Paddle Graph Learning (PGL) is an efficient and.. Apache-2

Karate Club (🥈23 · ⭐ 2.2K · 💤) - Karate Club: An API Oriented Open-source Python Framework.. ❗️GPL-3.0

jraph (🥈23 · ⭐ 1.4K · 💀) - A Graph Neural Network Library in Jax. Apache-2

graph4nlp (🥉22 · ⭐ 1.7K · 💀) - Graph4nlp is the library for the easy use of Graph.. Apache-2

graph-nets (🥉21 · ⭐ 5.4K · 💀) - Build Graph Nets in Tensorflow. Apache-2

pyRDF2Vec (🥉21 · ⭐ 250 · 💀) - Python Implementation and Extension of RDF2Vec. MIT

DeepWalk (🥉20 · ⭐ 2.7K · 💀) - DeepWalk - Deep Learning for Graphs. ❗️GPL-3.0

DIG (🥉20 · ⭐ 1.9K · 💤) - A library for graph deep learning research. ❗️GPL-3.0

deepsnap (🥉20 · ⭐ 560 · 💀) - Python library assists deep learning on graphs. MIT

GraphGym (🥉19 · ⭐ 1.7K · 💀) - Platform for designing and evaluating Graph Neural Networks (GNN). MIT

DeepGraph (🥉18 · ⭐ 290 · 💤) - Analyze Data with Pandas-based Networks. Documentation:. BSD-3

kglib (🥉17 · ⭐ 550 · 💀) - TypeDB-ML is the Machine Learning integrations library for TypeDB. Apache-2

Sematch (🥉17 · ⭐ 440 · 💀) - semantic similarity framework for knowledge graph. Apache-2

Euler (🥉16 · ⭐ 2.9K · 💀) - A distributed graph deep learning framework. Apache-2

ptgnn (🥉15 · ⭐ 370 · 💀) - A PyTorch Graph Neural Network Library. MIT

GraphEmbedding (🥉14 · ⭐ 3.8K · 💀) - Implementation and experiments of graph embedding.. MIT

GraphSAGE (🥉14 · ⭐ 3.5K · 💀) - Representation learning on large graphs using stochastic.. MIT

OpenNE (🥉14 · ⭐ 1.7K · 💀) - An Open-Source Package for Network Embedding (NE). MIT

OpenKE (🥉13 · ⭐ 3.9K · 💀) - An Open-Source Package for Knowledge Embedding (KE). ❗Unlicensed

Audio Data

Libraries for audio analysis, manipulation, transformation, and extraction, as well as speech recognition and music generation tasks.

speechbrain (🥇40 · ⭐ 9.3K) - A PyTorch-based Speech Toolkit. Apache-2

GitHub (👨‍💻 260 · 🔀 1.4K · 📦 2.8K · 📋 1.2K - 12% open · ⏱️ 11.02.2025):
```
	git clone https://github.com/speechbrain/speechbrain
```
PyPi (📥 3.4M / month · 📦 67 · ⏱️ 30.10.2024):
```
	pip install speechbrain
```

espnet (🥇38 · ⭐ 8.8K) - End-to-End Speech Processing Toolkit. Apache-2

GitHub (👨‍💻 490 · 🔀 2.2K · 📥 84 · 📦 410 · 📋 2.5K - 14% open · ⏱️ 05.02.2025):

	git clone https://github.com/espnet/espnet

PyPi (📥 21K / month · 📦 12 · ⏱️ 04.12.2024):

	pip install espnet

SpeechRecognition (🥇36 · ⭐ 8.6K) - Speech recognition module for Python, supporting several.. BSD-3

GitHub (👨‍💻 53 · 🔀 2.4K · 📦 21 · 📋 660 - 48% open · ⏱️ 30.01.2025):

	git clone https://github.com/Uberi/speech_recognition

PyPi (📥 1.3M / month · 📦 650 · ⏱️ 25.01.2025):

	pip install SpeechRecognition

Conda (📥 220K · ⏱️ 25.01.2025):

	conda install -c conda-forge speechrecognition

Coqui TTS (🥈35 · ⭐ 38K · 💤) - - a deep learning toolkit for Text-to-Speech, battle-.. MPL-2.0

GitHub (👨‍💻 170 · 🔀 4.5K · 📥 4.4M · 📦 2.3K · 📋 1.2K - 7% open · ⏱️ 10.02.2024):

	git clone https://github.com/coqui-ai/TTS

PyPi (📥 180K / month · 📦 53 · ⏱️ 12.12.2023):

	pip install tts

Conda (📥 22K · ⏱️ 16.06.2023):

	conda install -c conda-forge tts

torchaudio (🥈35 · ⭐ 2.6K) - Data manipulation and transformation for audio signal.. BSD-2

GitHub (👨‍💻 230 · 🔀 670 · 📋 1K - 27% open · ⏱️ 07.02.2025):

	git clone https://github.com/pytorch/audio

PyPi (📥 7.1M / month · 📦 1.6K · ⏱️ 29.01.2025):

	pip install torchaudio

spleeter (🥈33 · ⭐ 26K) - Deezer source separation library including pretrained models. MIT

GitHub (👨‍💻 22 · 🔀 2.9K · 📥 3.8M · 📦 900 · 📋 800 - 29% open · ⏱️ 24.01.2025):

	git clone https://github.com/deezer/spleeter

PyPi (📥 25K / month · 📦 12 · ⏱️ 10.06.2022):

	pip install spleeter

Conda (📥 100K · ⏱️ 16.06.2023):

	conda install -c conda-forge spleeter

Magenta (🥈32 · ⭐ 19K) - Magenta: Music and Art Generation with Machine Intelligence. Apache-2

GitHub (👨‍💻 160 · 🔀 3.7K · 📦 550 · 📋 1K - 41% open · ⏱️ 17.01.2025):

	git clone https://github.com/magenta/magenta

PyPi (📥 11K / month · 📦 5 · ⏱️ 01.08.2022):

	pip install magenta

librosa (🥈32 · ⭐ 7.4K) - Python library for audio and music analysis. ISC

GitHub (👨‍💻 120 · 🔀 970 · 📋 1.2K - 4% open · ⏱️ 11.02.2025):

	git clone https://github.com/librosa/librosa

PyPi (📥 2.9M / month · 📦 1.6K · ⏱️ 11.02.2025):

	pip install librosa

Conda (📥 860K · ⏱️ 19.12.2024):

	conda install -c conda-forge librosa

python-soundfile (🥈32 · ⭐ 730) - SoundFile is an audio library based on libsndfile, CFFI, and.. BSD-3

GitHub (👨‍💻 37 · 🔀 110 · 📥 21K · 📦 52K · 📋 260 - 46% open · ⏱️ 25.01.2025):

	git clone https://github.com/bastibe/python-soundfile

PyPi (📥 4.8M / month · 📦 1.1K · ⏱️ 25.01.2025):

	pip install soundfile

Conda:

	conda install -c anaconda pysoundfile

Porcupine (🥈31 · ⭐ 3.9K) - On-device wake word detection powered by deep learning. Apache-2

GitHub (👨‍💻 42 · 🔀 510 · 📦 40 · 📋 560 - 0% open · ⏱️ 12.02.2025):

	git clone https://github.com/Picovoice/Porcupine

PyPi (📥 19K / month · 📦 38 · ⏱️ 05.02.2025):

	pip install pvporcupine

audiomentations (🥉29 · ⭐ 1.9K) - A Python library for audio data augmentation. Inspired by.. MIT

GitHub (👨‍💻 30 · 🔀 190 · 📦 670 · 📋 190 - 25% open · ⏱️ 12.02.2025):

	git clone https://github.com/iver56/audiomentations

PyPi (📥 56K / month · 📦 25 · ⏱️ 12.02.2025):

	pip install audiomentations

Madmom (🥉27 · ⭐ 1.4K) - Python audio and music signal processing library. BSD-3

GitHub (👨‍💻 24 · 🔀 200 · 📦 460 · 📋 280 - 24% open · ⏱️ 25.08.2024):

	git clone https://github.com/CPJKU/madmom

PyPi (📥 2.5K / month · 📦 27 · ⏱️ 14.11.2018):

	pip install madmom

tinytag (🥉27 · ⭐ 730) - Python library for reading audio file metadata. MIT

GitHub (👨‍💻 27 · 🔀 100 · 📦 1.1K · 📋 120 - 3% open · ⏱️ 26.01.2025):

	git clone https://github.com/devsnd/tinytag

PyPi (📥 50K / month · 📦 110 · ⏱️ 03.11.2024):

	pip install tinytag

DDSP (🥉24 · ⭐ 3K) - DDSP: Differentiable Digital Signal Processing. Apache-2

GitHub (👨‍💻 32 · 🔀 340 · 📦 62 · 📋 170 - 28% open · ⏱️ 23.09.2024):

	git clone https://github.com/magenta/ddsp

PyPi (📥 3.8K / month · 📦 1 · ⏱️ 25.05.2022):

	pip install ddsp

Conda (📥 21K · ⏱️ 16.06.2023):

	conda install -c conda-forge ddsp

nnAudio (🥉22 · ⭐ 1.1K · 💤) - Audio processing by using pytorch 1D convolution network. MIT

GitHub (👨‍💻 15 · 🔀 91 · 📦 280 · 📋 63 - 28% open · ⏱️ 13.02.2024):

	git clone https://github.com/KinWaiCheuk/nnAudio

PyPi (📥 43K / month · 📦 4 · ⏱️ 13.02.2024):

	pip install nnAudio

Show 14 hidden projects...

Pydub (🥇36 · ⭐ 9.2K · 💀) - Manipulate audio with a simple and easy high level interface. MIT

DeepSpeech (🥈33 · ⭐ 26K · 💀) - DeepSpeech is an open source embedded (offline, on-.. MPL-2.0

audioread (🥉29 · ⭐ 500 · 💀) - cross-library (GStreamer + Core Audio + MAD + FFmpeg) audio.. MIT

pyAudioAnalysis (🥉28 · ⭐ 6K · 💀) - Python Audio Analysis Library: Feature Extraction,.. Apache-2

Essentia (🥉28 · ⭐ 3K) - C++ library for audio and music analysis, description and.. ❗️AGPL-3.0

aubio (🥉27 · ⭐ 3.4K · 💀) - a library for audio and music analysis. ❗️GPL-3.0

TTS (🥉26 · ⭐ 9.6K · 💀) - Deep learning for Text to Speech (Discussion forum:.. MPL-2.0

python_speech_features (🥉26 · ⭐ 2.4K · 💀) - This library provides common speech features for ASR.. MIT

Dejavu (🥉23 · ⭐ 6.5K · 💀) - Audio fingerprinting and recognition in Python. MIT

kapre (🥉22 · ⭐ 920 · 💀) - kapre: Keras Audio Preprocessors. MIT

Julius (🥉21 · ⭐ 430 · 💀) - Fast PyTorch based DSP for audio and 1D signals. MIT

TimeSide (🥉21 · ⭐ 380) - scalable audio processing framework and server written in Python. ❗️AGPL-3.0

Muda (🥉18 · ⭐ 230 · 💀) - A library for augmenting annotated audio data. ISC

textlesslib (🥉10 · ⭐ 530 · 💀) - Library for Textless Spoken Language Processing. MIT

Geospatial Data

Libraries to load, process, analyze, and write geographic data as well as libraries for spatial analysis, map visualization, and geocoding.

pydeck (🥇43 · ⭐ 12K) - WebGL2 powered visualization framework. MIT

GitHub (👨‍💻 280 · 🔀 2.1K · 📦 8.7K · 📋 3.1K - 11% open · ⏱️ 12.02.2025):
```
	git clone https://github.com/visgl/deck.gl
```
PyPi (📥 6.2M / month · 📦 120 · ⏱️ 10.05.2024):
```
	pip install pydeck
```
Conda (📥 690K · ⏱️ 31.01.2025):
```
	conda install -c conda-forge pydeck
```
npm (📥 550K / month · 📦 320 · ⏱️ 11.02.2025):
```
	npm install deck.gl
```

Shapely (🥇40 · ⭐ 4K) - Manipulation and analysis of geometric objects. BSD-3

GitHub (👨‍💻 160 · 🔀 570 · 📥 3.8K · 📦 94K · 📋 1.3K - 23% open · ⏱️ 10.02.2025):

	git clone https://github.com/shapely/shapely

PyPi (📥 38M / month · 📦 3.6K · ⏱️ 31.01.2025):

	pip install shapely

Conda (📥 11M · ⏱️ 31.01.2025):

	conda install -c conda-forge shapely

folium (🥇39 · ⭐ 7K) - Python Data. Leaflet.js Maps. MIT

GitHub (👨‍💻 170 · 🔀 2.2K · 📦 51K · 📋 1.1K - 7% open · ⏱️ 03.02.2025):

	git clone https://github.com/python-visualization/folium

PyPi (📥 1.7M / month · 📦 820 · ⏱️ 06.01.2025):

	pip install folium

Conda (📥 3.4M · ⏱️ 07.01.2025):

	conda install -c conda-forge folium

GeoPandas (🥈38 · ⭐ 4.6K) - Python tools for geographic data. BSD-3

GitHub (👨‍💻 240 · 🔀 940 · 📥 2.9K · 📦 48K · 📋 1.7K - 25% open · ⏱️ 13.02.2025):

	git clone https://github.com/geopandas/geopandas

PyPi (📥 6.5M / month · 📦 2.8K · ⏱️ 02.07.2024):

	pip install geopandas

Conda (📥 4.4M · ⏱️ 16.12.2024):

	conda install -c conda-forge geopandas

Rasterio (🥈36 · ⭐ 2.3K · 📉) - Rasterio reads and writes geospatial raster datasets. BSD-3

GitHub (👨‍💻 160 · 🔀 540 · 📥 1K · 📦 15K · 📋 1.9K - 7% open · ⏱️ 17.01.2025):

	git clone https://github.com/rasterio/rasterio

PyPi (📥 2.3M / month · 📦 1.5K · ⏱️ 02.12.2024):

	pip install rasterio

Conda (📥 4.3M · ⏱️ 02.12.2024):

	conda install -c conda-forge rasterio

Fiona (🥈36 · ⭐ 1.2K) - Fiona reads and writes geographic data files. BSD-3

GitHub (👨‍💻 77 · 🔀 210 · 📦 24K · 📋 810 - 4% open · ⏱️ 07.01.2025):

	git clone https://github.com/Toblerity/Fiona

PyPi (📥 4.4M / month · 📦 300 · ⏱️ 16.09.2024):

	pip install fiona

Conda (📥 6.5M · ⏱️ 06.12.2024):

	conda install -c conda-forge fiona

ArcGIS API (🥈35 · ⭐ 1.9K) - Documentation and samples for ArcGIS API for Python. Apache-2

GitHub (👨‍💻 96 · 🔀 1.1K · 📥 14K · 📦 900 · 📋 810 - 7% open · ⏱️ 10.02.2025):

	git clone https://github.com/Esri/arcgis-python-api

PyPi (📥 83K / month · 📦 40 · ⏱️ 01.10.2024):

	pip install arcgis

Docker Hub:

	docker pull esridocker/arcgis-api-python-notebook

pyproj (🥈35 · ⭐ 1.1K) - Python interface to PROJ (cartographic projections and coordinate.. MIT

GitHub (👨‍💻 69 · 🔀 220 · 📦 39K · 📋 630 - 5% open · ⏱️ 04.12.2024):

	git clone https://github.com/pyproj4/pyproj

PyPi (📥 8.9M / month · 📦 1.7K · ⏱️ 01.10.2024):

	pip install pyproj

Conda (📥 9.7M · ⏱️ 01.10.2024):

	conda install -c conda-forge pyproj

ipyleaflet (🥉31 · ⭐ 1.5K) - A Jupyter - Leaflet.js bridge. MIT

GitHub (👨‍💻 91 · 🔀 360 · 📦 14K · 📋 660 - 45% open · ⏱️ 05.12.2024):

	git clone https://github.com/jupyter-widgets/ipyleaflet

PyPi (📥 170K / month · 📦 280 · ⏱️ 22.07.2024):

	pip install ipyleaflet

Conda (📥 1.4M · ⏱️ 16.12.2024):

	conda install -c conda-forge ipyleaflet

npm (📥 5.4K / month · 📦 9 · ⏱️ 22.07.2024):

	npm install jupyter-leaflet

PySAL (🥉31 · ⭐ 1.4K) - PySAL: Python Spatial Analysis Library Meta-Package. BSD-3

GitHub (👨‍💻 79 · 🔀 300 · 📦 1.7K · 📋 660 - 2% open · ⏱️ 06.02.2025):

	git clone https://github.com/pysal/pysal

PyPi (📥 31K / month · 📦 59 · ⏱️ 06.02.2025):

	pip install pysal

Conda (📥 600K · ⏱️ 06.02.2025):

	conda install -c conda-forge pysal

geojson (🥉30 · ⭐ 940) - Python bindings and utilities for GeoJSON. BSD-3

GitHub (👨‍💻 58 · 🔀 120 · 📦 19K · 📋 100 - 23% open · ⏱️ 21.12.2024):

	git clone https://github.com/jazzband/geojson

PyPi (📥 2.7M / month · 📦 720 · ⏱️ 21.12.2024):

	pip install geojson

Conda (📥 910K · ⏱️ 22.12.2024):

	conda install -c conda-forge geojson

GeoViews (🥉29 · ⭐ 610) - Simple, concise geographical visualization in Python. BSD-3

GitHub (👨‍💻 33 · 🔀 77 · 📦 1.2K · 📋 350 - 30% open · ⏱️ 13.02.2025):

	git clone https://github.com/holoviz/geoviews

PyPi (📥 17K / month · 📦 63 · ⏱️ 17.12.2024):

	pip install geoviews

Conda (📥 280K · ⏱️ 18.12.2024):

	conda install -c conda-forge geoviews

pymap3d (🥉25 · ⭐ 400) - pure-Python (Numpy optional) 3D coordinate conversions for geospace ecef.. BSD-2

GitHub (👨‍💻 19 · 🔀 87 · 📦 470 · 📋 60 - 15% open · ⏱️ 08.01.2025):

	git clone https://github.com/geospace-code/pymap3d

PyPi (📥 270K / month · 📦 44 · ⏱️ 11.02.2024):

	pip install pymap3d

Conda (📥 93K · ⏱️ 25.12.2024):

	conda install -c conda-forge pymap3d

Mapbox GL (🥉24 · ⭐ 670) - Use Mapbox GL JS to visualize data in a Python Jupyter notebook. MIT

GitHub (👨‍💻 23 · 🔀 140 · 📦 240 · 📋 110 - 38% open · ⏱️ 06.02.2025):

	git clone https://github.com/mapbox/mapboxgl-jupyter

PyPi (📥 13K / month · 📦 12 · ⏱️ 02.06.2019):

	pip install mapboxgl

Show 8 hidden projects...

geopy (🥈33 · ⭐ 4.6K · 💀) - Geocoding library for Python. MIT

Geocoder (🥈33 · ⭐ 1.6K · 💀) - Python Geocoder. MIT

Satpy (🥉31 · ⭐ 1.1K) - Python package for earth-observing satellite data processing. ❗️GPL-3.0

Sentinelsat (🥉27 · ⭐ 990 · 💤) - Search and download Copernicus Sentinel satellite images. ❗️GPL-3.0

EarthPy (🥉26 · ⭐ 520 · 💀) - A package built to support working with spatial data using open.. BSD-3

prettymaps (🥉25 · ⭐ 12K) - A small set of Python functions to draw pretty maps from.. ❗️AGPL-3.0

gmaps (🥉22 · ⭐ 760 · 💀) - Google maps for Jupyter notebooks. BSD-3

geoplotlib (🥉20 · ⭐ 1K · 💀) - python toolbox for visualizing geographical data and making maps. MIT

Financial Data

Libraries for algorithmic stock/crypto trading, risk analytics, backtesting, technical analysis, and other tasks on financial data.

yfinance (🥇42 · ⭐ 16K) - Download market data from Yahoo! Finances API. Apache-2

GitHub (👨‍💻 130 · 🔀 2.5K · 📦 61K · 📋 1.5K - 13% open · ⏱️ 18.01.2025):
```
	git clone https://github.com/ranaroussi/yfinance
```
PyPi (📥 4.5M / month · 📦 800 · ⏱️ 18.01.2025):
```
	pip install yfinance
```
Conda (📥 98K · ⏱️ 16.06.2023):
```
	conda install -c ranaroussi yfinance
```

Qlib (🥈31 · ⭐ 16K) - Qlib is an AI-oriented quantitative investment platform that aims to.. MIT

GitHub (👨‍💻 140 · 🔀 2.7K · 📥 780 · 📦 21 · 📋 950 - 26% open · ⏱️ 09.01.2025):

	git clone https://github.com/microsoft/qlib

PyPi (📥 6.7K / month · 📦 1 · ⏱️ 23.12.2024):

	pip install pyqlib

bt (🥈30 · ⭐ 2.4K) - bt - flexible backtesting for Python. MIT

GitHub (👨‍💻 34 · 🔀 440 · 📦 1.6K · 📋 350 - 23% open · ⏱️ 12.02.2025):

	git clone https://github.com/pmorissette/bt

PyPi (📥 6.4K / month · 📦 13 · ⏱️ 12.02.2025):

	pip install bt

Conda (📥 71K · ⏱️ 21.09.2024):

	conda install -c conda-forge bt

ffn (🥈29 · ⭐ 2.1K) - ffn - a financial function library for Python. MIT

GitHub (👨‍💻 36 · 🔀 310 · 📦 520 · 📋 130 - 16% open · ⏱️ 12.02.2025):

	git clone https://github.com/pmorissette/ffn

PyPi (📥 16K / month · 📦 22 · ⏱️ 11.02.2025):

	pip install ffn

Conda (📥 16K · ⏱️ 12.02.2025):

	conda install -c conda-forge ffn

IB-insync (🥈27 · ⭐ 2.9K · 💤) - Python sync/async framework for Interactive Brokers API. BSD-2

GitHub (👨‍💻 36 · 🔀 800 · 📋 590 - 3% open · ⏱️ 14.03.2024):

	git clone https://github.com/erdewit/ib_insync

PyPi (📥 44K / month · 📦 44 · ⏱️ 21.11.2022):

	pip install ib_insync

Conda (📥 53K · ⏱️ 16.06.2023):

	conda install -c conda-forge ib-insync

stockstats (🥈27 · ⭐ 1.3K) - Supply a wrapper ``StockDataFrame`` based on the.. BSD-3

GitHub (👨‍💻 10 · 🔀 300 · 📦 1.2K · 📋 130 - 11% open · ⏱️ 02.02.2025):

	git clone https://github.com/jealous/stockstats

PyPi (📥 12K / month · 📦 12 · ⏱️ 02.02.2025):

	pip install stockstats

TensorTrade (🥉26 · ⭐ 4.6K · 💤) - An open source reinforcement learning framework for.. Apache-2

GitHub (👨‍💻 61 · 🔀 1K · 📦 67 · 📋 260 - 20% open · ⏱️ 09.06.2024):

	git clone https://github.com/tensortrade-org/tensortrade

PyPi (📥 2.3K / month · 📦 1 · ⏱️ 10.05.2021):

	pip install tensortrade

Conda (📥 4.5K · ⏱️ 16.06.2023):

	conda install -c conda-forge tensortrade

Alpha Vantage (🥉26 · ⭐ 4.4K · 💤) - A python wrapper for Alpha Vantage API for financial data. MIT

GitHub (👨‍💻 44 · 🔀 740 · 📋 290 - 0% open · ⏱️ 18.07.2024):

	git clone https://github.com/RomelTorres/alpha_vantage

PyPi (📥 51K / month · 📦 35 · ⏱️ 18.07.2024):

	pip install alpha_vantage

Conda (📥 8.4K · ⏱️ 09.08.2024):

	conda install -c conda-forge alpha_vantage

tf-quant-finance (🥉22 · ⭐ 4.7K) - High-performance TensorFlow library for quantitative.. Apache-2

GitHub (👨‍💻 48 · 🔀 580 · 📋 63 - 55% open · ⏱️ 10.02.2025):

	git clone https://github.com/google/tf-quant-finance

PyPi (📥 1.1K / month · 📦 3 · ⏱️ 19.08.2022):

	pip install tf-quant-finance

finmarketpy (🥉21 · ⭐ 3.5K) - Python library for backtesting trading strategies & analyzing.. Apache-2

GitHub (👨‍💻 18 · 🔀 510 · 📥 57 · 📦 16 · 📋 30 - 86% open · ⏱️ 09.11.2024):

	git clone https://github.com/cuemacro/finmarketpy

PyPi (📥 490 / month · ⏱️ 19.05.2024):

	pip install finmarketpy

Show 15 hidden projects...

zipline (🥇32 · ⭐ 18K · 💀) - Zipline, a Pythonic Algorithmic Trading Library. Apache-2

arch (🥇32 · ⭐ 1.4K) - ARCH models in Python. ❗Unlicensed

pyfolio (🥈31 · ⭐ 5.8K · 💀) - Portfolio and risk analytics in Python. Apache-2

ta (🥈31 · ⭐ 4.5K · 💀) - Technical Analysis Library using Pandas and Numpy. MIT

backtrader (🥈29 · ⭐ 16K · 💀) - Python Backtesting library for trading strategies. ❗️GPL-3.0

Backtesting.py (🥈27 · ⭐ 5.9K) - Backtest trading strategies in Python. ❗️AGPL-3.0

Alphalens (🥈27 · ⭐ 3.5K · 💀) - Performance analysis of predictive (alpha) stock factors. Apache-2

empyrical (🥈27 · ⭐ 1.3K · 💀) - Common financial risk and performance metrics. Used by.. Apache-2

Enigma Catalyst (🥉26 · ⭐ 2.5K · 💀) - An Algorithmic Trading Library for Crypto-Assets in.. Apache-2

PyAlgoTrade (🥉24 · ⭐ 4.5K · 💀) - Python Algorithmic Trading Library. Apache-2

FinTA (🥉24 · ⭐ 2.2K · 💀) - Common financial technical indicators implemented in Pandas. ❗️LGPL-3.0

Crypto Signals (🥉22 · ⭐ 5.1K · 💀) - Github.com/CryptoSignal - Trading & Technical Analysis Bot -.. MIT

FinQuant (🥉22 · ⭐ 1.5K · 💀) - A program for financial portfolio management, analysis and.. MIT

surpriver (🥉12 · ⭐ 1.8K · 💀) - Find big moving stocks before they move using machine.. ❗️GPL-3.0

pyrtfolio (🥉10 · ⭐ 150 · 💀) - Python package to generate stock portfolios. ❗️GPL-3.0

Time Series Data

Libraries for forecasting, anomaly detection, feature extraction, and machine learning on time-series and sequential data.

sktime (🥇40 · ⭐ 8.2K) - A unified framework for machine learning with time series. BSD-3

GitHub (👨‍💻 440 · 🔀 1.4K · 📥 110 · 📦 3.9K · 📋 2.7K - 38% open · ⏱️ 13.02.2025):
```
	git clone https://github.com/alan-turing-institute/sktime
```
PyPi (📥 860K / month · 📦 130 · ⏱️ 05.02.2025):
```
	pip install sktime
```

Conda (📥 1.1M · ⏱️ 04.01.2025):

	conda install -c conda-forge sktime-all-extras

StatsForecast (🥇34 · ⭐ 4.1K) - Lightning fast forecasting with statistical and econometric.. Apache-2

GitHub (👨‍💻 49 · 🔀 300 · 📦 1.4K · 📋 360 - 28% open · ⏱️ 12.02.2025):

	git clone https://github.com/Nixtla/statsforecast

PyPi (📥 990K / month · 📦 59 · ⏱️ 26.11.2024):

	pip install statsforecast

Conda (📥 150K · ⏱️ 06.02.2025):

	conda install -c conda-forge statsforecast

Prophet (🥇33 · ⭐ 19K) - Tool for producing high quality forecasts for time series data that has.. MIT

GitHub (👨‍💻 180 · 🔀 4.5K · 📥 2.9K · 📦 21 · 📋 2.2K - 19% open · ⏱️ 20.10.2024):

	git clone https://github.com/facebook/prophet

PyPi (📥 180K / month · 📦 91 · ⏱️ 05.09.2020):

	pip install fbprophet

Conda (📥 1.4M · ⏱️ 04.10.2024):

	conda install -c conda-forge prophet

Darts (🥈32 · ⭐ 8.3K) - A python library for user-friendly forecasting and anomaly detection.. Apache-2

GitHub (👨‍💻 130 · 🔀 900 · 📋 1.6K - 15% open · ⏱️ 08.02.2025):

	git clone https://github.com/unit8co/darts

PyPi (📥 88K / month · 📦 10 · ⏱️ 21.12.2024):

	pip install u8darts

Conda (📥 70K · ⏱️ 21.12.2024):

	conda install -c conda-forge u8darts-all

Docker Hub (📥 1.2K · ⏱️ 21.12.2024):

	docker pull unit8/darts

pytorch-forecasting (🥈32 · ⭐ 4.1K) - Time series forecasting with PyTorch. MIT

GitHub (👨‍💻 63 · 🔀 650 · 📦 500 · 📋 820 - 60% open · ⏱️ 06.02.2025):

	git clone https://github.com/jdb78/pytorch-forecasting

PyPi (📥 82K / month · 📦 22 · ⏱️ 06.02.2025):

	pip install pytorch-forecasting

Conda (📥 72K · ⏱️ 12.01.2025):

	conda install -c conda-forge pytorch-forecasting

STUMPY (🥈32 · ⭐ 3.8K) - STUMPY is a powerful and scalable Python library for modern time series.. BSD-3

GitHub (👨‍💻 41 · 🔀 330 · 📦 1.1K · 📋 530 - 13% open · ⏱️ 01.02.2025):

	git clone https://github.com/TDAmeritrade/stumpy

PyPi (📥 300K / month · 📦 30 · ⏱️ 09.07.2024):

	pip install stumpy

Conda (📥 1.1M · ⏱️ 21.12.2024):

	conda install -c conda-forge stumpy

NeuralForecast (🥈32 · ⭐ 3.3K) - Scalable and user friendly neural forecasting algorithms. Apache-2

GitHub (👨‍💻 49 · 🔀 380 · 📦 300 · 📋 590 - 20% open · ⏱️ 10.02.2025):

	git clone https://github.com/Nixtla/neuralforecast

PyPi (📥 82K / month · 📦 25 · ⏱️ 22.01.2025):

	pip install neuralforecast

Conda (📥 31K · ⏱️ 23.01.2025):

	conda install -c conda-forge neuralforecast

tsfresh (🥈31 · ⭐ 8.6K) - Automatic extraction of relevant features from time series:. MIT

GitHub (👨‍💻 98 · 🔀 1.2K · 📦 21 · 📋 540 - 12% open · ⏱️ 25.01.2025):

	git clone https://github.com/blue-yonder/tsfresh

PyPi (📥 250K / month · 📦 93 · ⏱️ 03.08.2024):

	pip install tsfresh

Conda (📥 1.4M · ⏱️ 11.01.2025):

	conda install -c conda-forge tsfresh

pmdarima (🥈31 · ⭐ 1.6K) - A statistical library designed to fill the void in Pythons time series.. MIT

GitHub (👨‍💻 23 · 🔀 240 · 📦 11K · 📋 340 - 19% open · ⏱️ 07.11.2024):

	git clone https://github.com/alkaline-ml/pmdarima

PyPi (📥 2.7M / month · 📦 150 · ⏱️ 23.10.2023):

	pip install pmdarima

Conda (📥 1.3M · ⏱️ 14.07.2024):

	conda install -c conda-forge pmdarima

GluonTS (🥈30 · ⭐ 4.8K) - Probabilistic time series modeling in Python. Apache-2

GitHub (👨‍💻 120 · 🔀 760 · 📋 970 - 34% open · ⏱️ 05.11.2024):

	git clone https://github.com/awslabs/gluon-ts

PyPi (📥 650K / month · 📦 33 · ⏱️ 11.11.2024):

	pip install gluonts

Conda (📥 1.4K · ⏱️ 16.12.2024):

	conda install -c anaconda gluonts

tslearn (🥈30 · ⭐ 2.9K · 💤) - The machine learning toolkit for time series analysis in Python. BSD-2

GitHub (👨‍💻 43 · 🔀 340 · 📦 1.6K · 📋 340 - 41% open · ⏱️ 01.07.2024):

	git clone https://github.com/tslearn-team/tslearn

PyPi (📥 380K / month · 📦 79 · ⏱️ 12.12.2023):

	pip install tslearn

Conda (📥 1.5M · ⏱️ 26.07.2024):

	conda install -c conda-forge tslearn

skforecast (🥈30 · ⭐ 1.2K) - Time series forecasting with machine learning models. BSD-3

GitHub (👨‍💻 17 · 🔀 150 · 📦 400 · 📋 190 - 13% open · ⏱️ 28.11.2024):

	git clone https://github.com/JoaquinAmatRodrigo/skforecast

PyPi (📥 110K / month · 📦 15 · ⏱️ 11.11.2024):

	pip install skforecast

NeuralProphet (🥉27 · ⭐ 4K) - NeuralProphet: A simple forecasting package. MIT

GitHub (👨‍💻 56 · 🔀 480 · 📋 560 - 10% open · ⏱️ 13.09.2024):

	git clone https://github.com/ourownstory/neural_prophet

PyPi (📥 64K / month · 📦 8 · ⏱️ 26.06.2024):

	pip install neuralprophet

Streamz (🥉27 · ⭐ 1.3K) - Real-time stream processing for python. BSD-3

GitHub (👨‍💻 49 · 🔀 150 · 📦 520 · 📋 270 - 44% open · ⏱️ 22.11.2024):

	git clone https://github.com/python-streamz/streamz

PyPi (📥 19K / month · 📦 57 · ⏱️ 27.07.2022):

	pip install streamz

Conda (📥 1.6M · ⏱️ 20.12.2024):

	conda install -c conda-forge streamz

TSFEL (🥉22 · ⭐ 970) - An intuitive library to extract features from time series. BSD-3

GitHub (👨‍💻 20 · 🔀 140 · 📦 170 · 📋 82 - 12% open · ⏱️ 17.10.2024):

	git clone https://github.com/fraunhoferportugal/tsfel

PyPi (📥 13K / month · 📦 7 · ⏱️ 12.09.2024):

	pip install tsfel

tsflex (🥉20 · ⭐ 410) - Flexible time series feature extraction & processing. MIT

GitHub (👨‍💻 6 · 🔀 26 · 📦 19 · 📋 56 - 58% open · ⏱️ 06.09.2024):

	git clone https://github.com/predict-idlab/tsflex

PyPi (📥 1.3K / month · 📦 2 · ⏱️ 06.09.2024):

	pip install tsflex

Conda (📥 30K · ⏱️ 08.04.2024):

	conda install -c conda-forge tsflex

pydlm (🥉19 · ⭐ 480) - A python library for Bayesian time series modeling. BSD-3

GitHub (👨‍💻 7 · 🔀 98 · 📦 39 · 📋 56 - 73% open · ⏱️ 07.09.2024):

	git clone https://github.com/wwrechard/pydlm

PyPi (📥 39K / month · 📦 2 · ⏱️ 13.08.2024):

	pip install pydlm

Auto TS (🥉18 · ⭐ 740 · 💤) - Automatically build ARIMA, SARIMAX, VAR, FB Prophet and XGBoost.. Apache-2

GitHub (👨‍💻 13 · 🔀 120 · 📋 90 - 2% open · ⏱️ 05.05.2024):

	git clone https://github.com/AutoViML/Auto_TS

PyPi (📥 9K / month · ⏱️ 05.05.2024):

	pip install auto-ts

Show 11 hidden projects...

pyts (🥉27 · ⭐ 1.8K · 💀) - A Python package for time series classification. BSD-3

PyFlux (🥉25 · ⭐ 2.1K · 💀) - Open source time series library for Python. BSD-3

luminol (🥉22 · ⭐ 1.2K · 💀) - Anomaly Detection and Correlation library. Apache-2

ADTK (🥉22 · ⭐ 1.1K · 💀) - A Python toolkit for rule-based/unsupervised anomaly detection in.. MPL-2.0

seglearn (🥉21 · ⭐ 570 · 💀) - Python module for machine learning time series:. BSD-3

tick (🥉21 · ⭐ 500 · 💀) - Module for statistical learning, with a particular emphasis on time-.. BSD-3

greykite (🥉20 · ⭐ 1.8K · 💀) - A flexible, intuitive and fast forecasting library. BSD-2

matrixprofile-ts (🥉19 · ⭐ 740 · 💀) - A Python library for detecting patterns and anomalies.. Apache-2

atspy (🥉15 · ⭐ 510 · 💀) - AtsPy: Automated Time Series Models in Python (by @firmai). MIT

tsaug (🥉14 · ⭐ 350 · 💀) - A Python package for time series augmentation. Apache-2

tslumen (🥉8 · ⭐ 69 · 💀) - A library for Time Series EDA (exploratory data analysis). Apache-2

Medical Data

Libraries for processing and analyzing medical data such as MRIs, EEGs, genomic data, and other medical imaging formats.

MNE (🥇39 · ⭐ 2.8K) - MNE: Magnetoencephalography (MEG) and Electroencephalography (EEG) in Python. BSD-3

GitHub (👨‍💻 390 · 🔀 1.3K · 📦 5.1K · 📋 5K - 11% open · ⏱️ 12.02.2025):
```
	git clone https://github.com/mne-tools/mne-python
```
PyPi (📥 150K / month · 📦 420 · ⏱️ 18.12.2024):
```
	pip install mne
```
Conda (📥 490K · ⏱️ 07.01.2025):
```
	conda install -c conda-forge mne
```

Nilearn (🥇39 · ⭐ 1.2K) - Machine learning for NeuroImaging in Python. BSD-3

GitHub (👨‍💻 250 · 🔀 600 · 📥 290 · 📦 3.8K · 📋 2.3K - 13% open · ⏱️ 12.02.2025):

	git clone https://github.com/nilearn/nilearn

PyPi (📥 76K / month · 📦 310 · ⏱️ 23.12.2024):

	pip install nilearn

Conda (📥 320K · ⏱️ 23.12.2024):

	conda install -c conda-forge nilearn

MONAI (🥈36 · ⭐ 6.1K) - AI Toolkit for Healthcare Imaging. Apache-2

GitHub (👨‍💻 220 · 🔀 1.1K · 📦 3.5K · 📋 3.2K - 12% open · ⏱️ 12.02.2025):

	git clone https://github.com/Project-MONAI/MONAI

PyPi (📥 230K / month · 📦 140 · ⏱️ 10.12.2024):

	pip install monai

Conda (📥 41K · ⏱️ 23.12.2024):

	conda install -c conda-forge monai

NIPYPE (🥈36 · ⭐ 760) - Workflows and interfaces for neuroimaging packages. Apache-2

GitHub (👨‍💻 260 · 🔀 530 · 📦 5.8K · 📋 1.4K - 30% open · ⏱️ 18.01.2025):

	git clone https://github.com/nipy/nipype

PyPi (📥 270K / month · 📦 150 · ⏱️ 17.12.2024):

	pip install nipype

Conda (📥 760K · ⏱️ 18.12.2024):

	conda install -c conda-forge nipype

NiBabel (🥈36 · ⭐ 680) - Python package to access a cacophony of neuro-imaging file formats. MIT

GitHub (👨‍💻 110 · 🔀 260 · 📦 25K · 📋 540 - 23% open · ⏱️ 16.01.2025):

	git clone https://github.com/nipy/nibabel

PyPi (📥 610K / month · 📦 1.2K · ⏱️ 23.10.2024):

	pip install nibabel

Conda (📥 850K · ⏱️ 12.12.2024):

	conda install -c conda-forge nibabel

Hail (🥈34 · ⭐ 1K) - Cloud-native genomic dataframes and batch computing. MIT

GitHub (👨‍💻 97 · 🔀 250 · 📦 160 · 📋 2.5K - 10% open · ⏱️ 11.02.2025):

	git clone https://github.com/hail-is/hail

PyPi (📥 160K / month · 📦 34 · ⏱️ 04.10.2024):

	pip install hail

Lifelines (🥈33 · ⭐ 2.4K) - Survival analysis in Python. MIT

GitHub (👨‍💻 120 · 🔀 560 · 📦 3.4K · 📋 980 - 27% open · ⏱️ 29.10.2024):

	git clone https://github.com/CamDavidsonPilon/lifelines

PyPi (📥 2.7M / month · 📦 160 · ⏱️ 29.10.2024):

	pip install lifelines

Conda (📥 400K · ⏱️ 19.12.2024):

	conda install -c conda-forge lifelines

DeepVariant (🥉27 · ⭐ 3.3K) - DeepVariant is an analysis pipeline that uses a deep neural.. BSD-3

GitHub (👨‍💻 36 · 🔀 730 · 📥 4.8K · 📋 870 - 0% open · ⏱️ 09.12.2024):

	git clone https://github.com/google/deepvariant

Conda (📥 74K · ⏱️ 16.06.2023):

	conda install -c bioconda deepvariant

Brainiak (🥉20 · ⭐ 350) - Brain Imaging Analysis Kit. Apache-2

GitHub (👨‍💻 35 · 🔀 140 · 📋 230 - 39% open · ⏱️ 06.01.2025):

	git clone https://github.com/brainiak/brainiak

PyPi (📥 2.6K / month · ⏱️ 07.01.2025):

	pip install brainiak

Docker Hub (📥 1.9K · ⭐ 1 · ⏱️ 07.01.2025):

	docker pull brainiak/brainiak

Show 10 hidden projects...

DIPY (🥈32 · ⭐ 730) - DIPY is the paragon 3D/4D+ medical imaging library in Python… ❗Unlicensed

NiftyNet (🥉25 · ⭐ 1.4K · 💀) - [unmaintained] An open-source convolutional neural.. Apache-2

NIPY (🥉25 · ⭐ 380) - Neuroimaging in Python FMRI analysis package. ❗Unlicensed

MedPy (🥉23 · ⭐ 580 · 💤) - Medical image processing in Python. ❗️GPL-3.0

Glow (🥉21 · ⭐ 270) - An open-source toolkit for large-scale genomic analysis. Apache-2

DLTK (🥉20 · ⭐ 1.4K · 💀) - Deep Learning Toolkit for Medical Image Analysis. Apache-2

MedicalTorch (🥉15 · ⭐ 860 · 💀) - A medical imaging framework for Pytorch. Apache-2

Medical Detection Toolkit (🥉14 · ⭐ 1.3K · 💀) - The Medical Detection Toolkit contains 2D + 3D.. Apache-2

DeepNeuro (🥉14 · ⭐ 120 · 💀) - A deep learning python package for neuroimaging data. Made by:. MIT

MedicalNet (🥉12 · ⭐ 2K · 💀) - Many studies have shown that the performance on deep learning is.. MIT

Tabular Data

Libraries for processing tabular and structured data.

pytorch_tabular (🥇24 · ⭐ 1.4K) - A standard framework for modelling Deep Learning Models.. MIT

GitHub (👨‍💻 25 · 🔀 150 · 📥 54 · 📋 160 - 6% open · ⏱️ 18.12.2024):
```
	git clone https://github.com/manujosephv/pytorch_tabular
```
PyPi (📥 4.9K / month · 📦 9 · ⏱️ 28.11.2024):
```
	pip install pytorch_tabular
```

miceforest (🥈23 · ⭐ 370) - Multiple Imputation with LightGBM in Python. MIT

GitHub (👨‍💻 8 · 🔀 30 · 📦 200 · 📋 86 - 9% open · ⏱️ 02.08.2024):

	git clone https://github.com/AnotherSamWilson/miceforest

PyPi (📥 64K / month · 📦 9 · ⏱️ 02.08.2024):

	pip install miceforest

Conda (📥 17K · ⏱️ 16.06.2023):

	conda install -c conda-forge miceforest

upgini (🥉21 · ⭐ 320) - Data search & enrichment library for Machine Learning Easily find and add.. BSD-3

GitHub (👨‍💻 13 · 🔀 25 · 📦 9 · ⏱️ 12.02.2025):

	git clone https://github.com/upgini/upgini

PyPi (📥 25K / month · ⏱️ 12.02.2025):

	pip install upgini

carefree-learn (🥉18 · ⭐ 410 · 💤) - Deep Learning PyTorch. MIT

GitHub (👨‍💻 1 · 🔀 38 · 📦 8 · 📋 82 - 2% open · ⏱️ 18.03.2024):

	git clone https://github.com/carefree0910/carefree-learn

PyPi (📥 1.3K / month · ⏱️ 09.01.2024):

	pip install carefree-learn

Show 1 hidden projects...

deltapy (🥉13 · ⭐ 540 · 💀) - DeltaPy - Tabular Data Augmentation (by @firmai). MIT

Optical Character Recognition

Libraries for optical character recognition (OCR) and text extraction from images or videos.

PaddleOCR (🥇41 · ⭐ 46K) - Awesome multilingual OCR toolkits based on PaddlePaddle.. Apache-2

GitHub (👨‍💻 270 · 🔀 7.9K · 📥 1.2M · 📦 4.3K · 📋 9.6K - 1% open · ⏱️ 11.02.2025):
```
	git clone https://github.com/PaddlePaddle/PaddleOCR
```
PyPi (📥 300K / month · 📦 110 · ⏱️ 22.10.2024):
```
	pip install paddleocr
```

OCRmyPDF (🥇37 · ⭐ 18K) - OCRmyPDF adds an OCR text layer to scanned PDF files, allowing them.. MPL-2.0

GitHub (👨‍💻 100 · 🔀 1.2K · 📥 7K · 📦 1.2K · 📋 1.2K - 9% open · ⏱️ 07.02.2025):

	git clone https://github.com/ocrmypdf/OCRmyPDF

PyPi (📥 200K / month · 📦 42 · ⏱️ 08.02.2025):

	pip install ocrmypdf

Conda (📥 90K · ⏱️ 16.06.2023):

	conda install -c conda-forge ocrmypdf

EasyOCR (🥈35 · ⭐ 26K) - Ready-to-use OCR with 80+ supported languages and all popular writing.. Apache-2

GitHub (👨‍💻 130 · 🔀 3.2K · 📥 18M · 📦 11K · 📋 1.1K - 43% open · ⏱️ 24.09.2024):

	git clone https://github.com/JaidedAI/EasyOCR

PyPi (📥 770K / month · 📦 250 · ⏱️ 24.09.2024):

	pip install easyocr

Tesseract (🥈32 · ⭐ 6K) - Python-tesseract is an optical character recognition (OCR) tool for.. Apache-2

GitHub (👨‍💻 49 · 🔀 720 · 📋 370 - 2% open · ⏱️ 04.02.2025):

	git clone https://github.com/madmaze/pytesseract

PyPi (📥 2.7M / month · 📦 970 · ⏱️ 16.08.2024):

	pip install pytesseract

Conda (📥 650K · ⏱️ 07.01.2025):

	conda install -c conda-forge pytesseract

tesserocr (🥈31 · ⭐ 2.1K · 📈) - A Python wrapper for the tesseract-ocr API. MIT

GitHub (👨‍💻 31 · 🔀 260 · 📥 820 · 📦 1.2K · 📋 280 - 17% open · ⏱️ 12.02.2025):

	git clone https://github.com/sirfz/tesserocr

PyPi (📥 96K / month · 📦 43 · ⏱️ 12.02.2025):

	pip install tesserocr

Conda (📥 230K · ⏱️ 13.09.2024):

	conda install -c conda-forge tesserocr

MMOCR (🥉26 · ⭐ 4.4K) - OpenMMLab Text Detection, Recognition and Understanding Toolbox. Apache-2

GitHub (👨‍💻 90 · 🔀 750 · 📦 200 · 📋 930 - 20% open · ⏱️ 27.11.2024):

	git clone https://github.com/open-mmlab/mmocr

PyPi (📥 4.9K / month · 📦 4 · ⏱️ 05.05.2022):

	pip install mmocr

Show 6 hidden projects...

keras-ocr (🥉25 · ⭐ 1.4K · 💀) - A packaged and flexible version of the CRAFT text detector.. MIT

calamari (🥉22 · ⭐ 1.1K) - Line based ATR Engine based on OCRopy. ❗️GPL-3.0

pdftabextract (🥉21 · ⭐ 2.2K · 💀) - A set of tools for extracting tables from PDF files.. Apache-2

attention-ocr (🥉21 · ⭐ 1.1K · 💀) - A Tensorflow model for text recognition (CNN + seq2seq.. MIT

doc2text (🥉20 · ⭐ 1.3K · 💀) - Detect text blocks and OCR poorly scanned PDFs in bulk. Python.. MIT

Mozart (🥉10 · ⭐ 640 · 💀) - An optical music recognition (OMR) system. Converts sheet.. Apache-2

Data Containers & Structures

General-purpose data containers & structures as well as utilities & extensions for pandas.

🔗 best-of-python - Data Containers ( ⭐ 3.8K) - Collection of data-container, dataframe, and pandas-..

Data Loading & Extraction

Libraries for loading, collecting, and extracting data from a variety of data sources and formats.

🔗 best-of-python - Data Extraction ( ⭐ 3.8K) - Collection of data-loading and -extraction libraries.

Web Scraping & Crawling

Libraries for web scraping, crawling, downloading, and mining as well as libraries.

🔗 best-of-web-python - Web Scraping ( ⭐ 2.4K · 💤) - Collection of web-scraping and crawling libraries.

Data Pipelines & Streaming

Libraries for data batch- and stream-processing, workflow automation, job scheduling, and other data pipeline tasks.

🔗 best-of-python - Data Pipelines ( ⭐ 3.8K) - Libraries for data batch- and stream-processing,..

Show 1 hidden projects...

pyclugen (🥇10 · ⭐ 8) - Multidimensional cluster generation in Python. MIT

Distributed Machine Learning

Libraries that provide capabilities to distribute and parallelize machine learning tasks across large-scale compute infrastructure.

Ray (🥇47 · ⭐ 35K) - Ray is an AI compute engine. Ray consists of a core distributed runtime.. Apache-2

GitHub (👨‍💻 1.2K · 🔀 6K · 📥 250 · 📦 21K · 📋 20K - 21% open · ⏱️ 13.02.2025):
```
	git clone https://github.com/ray-project/ray
```
PyPi (📥 6.2M / month · 📦 880 · ⏱️ 11.02.2025):
```
	pip install ray
```
Conda (📥 670K · ⏱️ 12.02.2025):
```
	conda install -c conda-forge ray-tune
```

dask (🥇44 · ⭐ 13K) - Parallel computing with task scheduling. BSD-3

GitHub (👨‍💻 620 · 🔀 1.7K · 📦 70K · 📋 5.5K - 20% open · ⏱️ 13.02.2025):

	git clone https://github.com/dask/dask

PyPi (📥 10M / month · 📦 2.6K · ⏱️ 17.01.2025):

	pip install dask

Conda (📥 13M · ⏱️ 19.01.2025):

	conda install -c conda-forge dask

DeepSpeed (🥇41 · ⭐ 37K) - DeepSpeed is a deep learning optimization library that makes.. Apache-2

GitHub (👨‍💻 370 · 🔀 4.2K · 📦 11K · 📋 3.1K - 35% open · ⏱️ 12.02.2025):

	git clone https://github.com/microsoft/DeepSpeed

PyPi (📥 620K / month · 📦 240 · ⏱️ 21.01.2025):

	pip install deepspeed

Docker Hub (📥 22K · ⭐ 4 · ⏱️ 02.09.2022):

	docker pull deepspeed/deepspeed

dask.distributed (🥇40 · ⭐ 1.6K) - A distributed task scheduler for Dask. BSD-3

GitHub (👨‍💻 330 · 🔀 720 · 📦 38K · 📋 4K - 38% open · ⏱️ 13.02.2025):

	git clone https://github.com/dask/distributed

PyPi (📥 4.3M / month · 📦 900 · ⏱️ 17.01.2025):

	pip install distributed

Conda (📥 16M · ⏱️ 19.01.2025):

	conda install -c conda-forge distributed

metrics (🥈36 · ⭐ 2.2K) - Machine learning metrics for distributed, scalable PyTorch.. Apache-2

GitHub (👨‍💻 270 · 🔀 420 · 📥 6.3K · 📦 37K · 📋 920 - 10% open · ⏱️ 13.02.2025):

	git clone https://github.com/Lightning-AI/metrics

PyPi (📥 5.1K / month · 📦 2 · ⏱️ 28.04.2018):

	pip install metrics

Conda (📥 1.8M · ⏱️ 26.12.2024):

	conda install -c conda-forge torchmetrics

horovod (🥈35 · ⭐ 14K) - Distributed training framework for TensorFlow, Keras, PyTorch, and.. Apache-2

GitHub (👨‍💻 170 · 🔀 2.2K · 📦 1.3K · 📋 2.3K - 17% open · ⏱️ 01.02.2025):

	git clone https://github.com/horovod/horovod

PyPi (📥 90K / month · 📦 33 · ⏱️ 12.06.2023):

	pip install horovod

H2O-3 (🥈34 · ⭐ 7K) - H2O is an Open Source, Distributed, Fast & Scalable Machine Learning.. Apache-2

GitHub (👨‍💻 270 · 🔀 2K · 📦 21 · 📋 9.6K - 29% open · ⏱️ 13.02.2025):

	git clone https://github.com/h2oai/h2o-3

PyPi (📥 190K / month · 📦 49 · ⏱️ 02.11.2024):

	pip install h2o

ColossalAI (🥈32 · ⭐ 39K) - Making large AI models cheaper, faster and more accessible. Apache-2

GitHub (👨‍💻 190 · 🔀 4.4K · 📦 460 · 📋 1.7K - 25% open · ⏱️ 12.02.2025):

	git clone https://github.com/hpcaitech/colossalai

BigDL (🥈32 · ⭐ 7.2K) - Accelerate local LLM inference and finetuning (LLaMA, Mistral,.. Apache-2

GitHub (👨‍💻 120 · 🔀 1.3K · 📥 710 · 📋 2.7K - 38% open · ⏱️ 13.02.2025):
```
	git clone https://github.com/intel-analytics/BigDL
```
PyPi (📥 20K / month · 📦 2 · ⏱️ 24.03.2024):
```
	pip install bigdl
```

Maven (📦 5 · ⏱️ 20.04.2021):

	<dependency>
		<groupId>com.intel.analytics.bigdl</groupId>
		<artifactId>bigdl-SPARK_2.4</artifactId>
		<version>[VERSION]</version>
	</dependency>

FairScale (🥈31 · ⭐ 3.3K) - PyTorch extensions for high performance and large scale training. BSD-3

GitHub (👨‍💻 76 · 🔀 280 · 📦 7.4K · 📋 390 - 26% open · ⏱️ 12.01.2025):

	git clone https://github.com/facebookresearch/fairscale

PyPi (📥 490K / month · 📦 150 · ⏱️ 11.12.2022):

	pip install fairscale

Conda (📥 400K · ⏱️ 28.11.2023):

	conda install -c conda-forge fairscale

dask-ml (🥈30 · ⭐ 920 · 📈) - Scalable Machine Learning with Dask. BSD-3

GitHub (👨‍💻 80 · 🔀 260 · 📦 1.2K · 📋 540 - 51% open · ⏱️ 07.02.2025):

	git clone https://github.com/dask/dask-ml

PyPi (📥 90K / month · 📦 100 · ⏱️ 08.02.2025):

	pip install dask-ml

Conda (📥 940K · ⏱️ 08.02.2025):

	conda install -c conda-forge dask-ml

SynapseML (🥈29 · ⭐ 5.1K) - Simple and Distributed Machine Learning. MIT

GitHub (👨‍💻 120 · 🔀 830 · 📋 790 - 48% open · ⏱️ 06.02.2025):

	git clone https://github.com/microsoft/SynapseML

PyPi (📥 270K / month · 📦 7 · ⏱️ 10.01.2025):

	pip install synapseml

mpi4py (🥈29 · ⭐ 830) - Python bindings for MPI. BSD-3

GitHub (👨‍💻 27 · 🔀 120 · 📥 30K · 📋 200 - 3% open · ⏱️ 12.02.2025):

	git clone https://github.com/mpi4py/mpi4py

PyPi (📥 360K / month · 📦 830 · ⏱️ 01.02.2025):

	pip install mpi4py

Conda (📥 3.5M · ⏱️ 02.02.2025):

	conda install -c conda-forge mpi4py

Submit it (🥉28 · ⭐ 1.4K) - Python 3.8+ toolbox for submitting jobs to Slurm. MIT

GitHub (👨‍💻 25 · 🔀 130 · 📦 3.9K · 📋 130 - 39% open · ⏱️ 18.09.2024):

	git clone https://github.com/facebookincubator/submitit

PyPi (📥 420K / month · 📦 49 · ⏱️ 18.09.2024):

	pip install submitit

Conda (📥 48K · ⏱️ 19.11.2024):

	conda install -c conda-forge submitit

Apache Singa (🥉24 · ⭐ 3.4K) - a distributed deep learning platform. Apache-2

GitHub (👨‍💻 96 · 🔀 1.2K · 📦 5 · 📋 140 - 36% open · ⏱️ 30.12.2024):

	git clone https://github.com/apache/singa

Conda (📥 940 · ⏱️ 16.06.2023):

	conda install -c nusdbsystem singa

Docker Hub (📥 8.7K · ⭐ 4 · ⏱️ 31.05.2022):

	docker pull apache/singa

Hivemind (🥉24 · ⭐ 2.1K) - Decentralized deep learning in PyTorch. Built to train models on.. MIT

GitHub (👨‍💻 33 · 🔀 170 · 📦 120 · 📋 180 - 43% open · ⏱️ 05.11.2024):

	git clone https://github.com/learning-at-home/hivemind

PyPi (📥 1.1K / month · 📦 10 · ⏱️ 31.08.2023):

	pip install hivemind

MMLSpark (🥉23 · ⭐ 5.1K) - Simple and Distributed Machine Learning. MIT

GitHub (👨‍💻 120 · 🔀 830 · 📋 790 - 48% open · ⏱️ 06.02.2025):

	git clone https://github.com/microsoft/SynapseML

PyPi (⏱️ 18.03.2020):

	pip install mmlspark

analytics-zoo (🥉23 · ⭐ 2.6K) - Distributed Tensorflow, Keras and PyTorch on Apache.. Apache-2

GitHub (👨‍💻 110 · 🔀 730 · 📋 1.3K - 32% open · ⏱️ 09.01.2025):

	git clone https://github.com/intel-analytics/analytics-zoo

PyPi (📥 1.9K / month · 📦 1 · ⏱️ 22.08.2022):

	pip install analytics-zoo

Show 18 hidden projects...

DEAP (🥈35 · ⭐ 6K) - Distributed Evolutionary Algorithms in Python. ❗️LGPL-3.0

ipyparallel (🥈29 · ⭐ 2.6K) - IPython Parallel: Interactive Parallel Computing in.. ❗Unlicensed

petastorm (🥉28 · ⭐ 1.8K · 💀) - Petastorm library enables single machine or distributed.. Apache-2

TensorFlowOnSpark (🥉26 · ⭐ 3.9K · 💀) - TensorFlowOnSpark brings TensorFlow programs to.. Apache-2

Elephas (🥉25 · ⭐ 1.6K · 💀) - Distributed Deep learning with Keras & Spark. MIT keras

BytePS (🥉22 · ⭐ 3.7K · 💀) - A high performance and generic framework for distributed DNN.. Apache-2

Mesh (🥉22 · ⭐ 1.6K · 💀) - Mesh TensorFlow: Model Parallelism Made Easier. Apache-2

sk-dist (🥉21 · ⭐ 280 · 💀) - Distributed scikit-learn meta-estimators in PySpark. Apache-2

somoclu (🥉20 · ⭐ 270 · 💀) - Massively parallel self-organizing maps: accelerate training on.. MIT

launchpad (🥉19 · ⭐ 320 · 💀) - Launchpad is a library that simplifies writing.. Apache-2

mesh-transformer-jax (🥉18 · ⭐ 6.3K · 💀) - Model parallel transformers in JAX and Haiku. Apache-2

bluefog (🥉18 · ⭐ 290 · 💀) - Distributed and decentralized training framework for PyTorch.. Apache-2

Fiber (🥉17 · ⭐ 1K · 💀) - Distributed Computing for AI Made Simple. Apache-2

parallelformers (🥉17 · ⭐ 780 · 💀) - Parallelformers: An Efficient Model Parallelization.. Apache-2

TensorFrames (🥉16 · ⭐ 720 · 💀) - Tensorflow wrapper for DataFrames on Apache Spark. Apache-2

LazyCluster (🥉15 · ⭐ 49 · 💀) - Distributed machine learning made simple. Apache-2

autodist (🥉12 · ⭐ 130 · 💀) - Simple Distributed Deep Learning on TensorFlow. Apache-2

moolib (🥉11 · ⭐ 370 · 💀) - A library for distributed ML training with PyTorch. MIT

Hyperparameter Optimization & AutoML

Libraries for hyperparameter optimization, automl and neural architecture search.

Optuna (🥇43 · ⭐ 11K) - A hyperparameter optimization framework. MIT

GitHub (👨‍💻 280 · 🔀 1.1K · 📦 22K · 📋 1.7K - 3% open · ⏱️ 13.02.2025):
```
	git clone https://github.com/optuna/optuna
```
PyPi (📥 3.5M / month · 📦 1.2K · ⏱️ 12.02.2025):
```
	pip install optuna
```
Conda (📥 2.2M · ⏱️ 20.01.2025):
```
	conda install -c conda-forge optuna
```

AutoGluon (🥇36 · ⭐ 8.4K) - Fast and Accurate ML in 3 Lines of Code. Apache-2

GitHub (👨‍💻 130 · 🔀 950 · 📦 970 · 📋 1.6K - 24% open · ⏱️ 12.02.2025):

	git clone https://github.com/autogluon/autogluon

PyPi (📥 230K / month · 📦 30 · ⏱️ 13.02.2025):

	pip install autogluon

Conda (📥 29K · ⏱️ 12.12.2024):

	conda install -c conda-forge autogluon

Docker Hub (📥 14K · ⭐ 17 · ⏱️ 07.03.2024):

	docker pull autogluon/autogluon

Ax (🥇36 · ⭐ 2.4K) - Adaptive Experimentation Platform. MIT

GitHub (👨‍💻 180 · 🔀 320 · 📦 890 · 📋 820 - 10% open · ⏱️ 13.02.2025):

	git clone https://github.com/facebook/Ax

PyPi (📥 140K / month · 📦 57 · ⏱️ 03.02.2025):

	pip install ax-platform

Conda (📥 34K · ⏱️ 07.02.2025):

	conda install -c conda-forge ax-platform

BoTorch (🥇34 · ⭐ 3.2K) - Bayesian optimization in PyTorch. MIT

GitHub (👨‍💻 140 · 🔀 410 · 📦 1.3K · 📋 570 - 13% open · ⏱️ 11.02.2025):

	git clone https://github.com/pytorch/botorch

PyPi (📥 240K / month · 📦 100 · ⏱️ 03.02.2025):

	pip install botorch

Conda (📥 140K · ⏱️ 04.02.2025):

	conda install -c conda-forge botorch

Hyperopt (🥇33 · ⭐ 7.3K) - Distributed Asynchronous Hyperparameter Optimization in Python. BSD-3

GitHub (👨‍💻 100 · 🔀 1.1K · 📦 19K · 📋 710 - 20% open · ⏱️ 27.12.2024):

	git clone https://github.com/hyperopt/hyperopt

PyPi (📥 2.2M / month · 📦 450 · ⏱️ 17.11.2021):

	pip install hyperopt

Conda (📥 810K · ⏱️ 20.12.2024):

	conda install -c conda-forge hyperopt

nevergrad (🥇33 · ⭐ 4K) - A Python toolbox for performing gradient-free optimization. MIT

GitHub (👨‍💻 57 · 🔀 360 · 📦 840 · 📋 310 - 39% open · ⏱️ 05.12.2024):

	git clone https://github.com/facebookresearch/nevergrad

PyPi (📥 110K / month · 📦 62 · ⏱️ 01.12.2024):

	pip install nevergrad

Conda (📥 59K · ⏱️ 09.01.2024):

	conda install -c conda-forge nevergrad

Bayesian Optimization (🥈32 · ⭐ 8.1K) - A Python implementation of global optimization with.. MIT

GitHub (👨‍💻 47 · 🔀 1.6K · 📥 170 · 📦 3.3K · 📋 370 - 1% open · ⏱️ 27.12.2024):

	git clone https://github.com/fmfn/BayesianOptimization

PyPi (📥 370K / month · 📦 150 · ⏱️ 27.12.2024):

	pip install bayesian-optimization

featuretools (🥈32 · ⭐ 7.4K) - An open source python library for automated feature engineering. BSD-3

GitHub (👨‍💻 74 · 🔀 880 · 📦 1.9K · 📋 1K - 15% open · ⏱️ 13.11.2024):

	git clone https://github.com/alteryx/featuretools

PyPi (📥 63K / month · 📦 74 · ⏱️ 14.05.2024):

	pip install featuretools

Conda (📥 230K · ⏱️ 15.05.2024):

	conda install -c conda-forge featuretools

AutoKeras (🥈31 · ⭐ 9.2K) - AutoML library for deep learning. Apache-2

GitHub (👨‍💻 140 · 🔀 1.4K · 📥 19K · 📦 800 · 📋 910 - 16% open · ⏱️ 16.12.2024):

	git clone https://github.com/keras-team/autokeras

PyPi (📥 19K / month · 📦 13 · ⏱️ 20.03.2024):

	pip install autokeras

Keras Tuner (🥈31 · ⭐ 2.9K · 💤) - A Hyperparameter Tuning Library for Keras. Apache-2

GitHub (👨‍💻 61 · 🔀 400 · 📦 5.2K · 📋 500 - 44% open · ⏱️ 24.06.2024):

	git clone https://github.com/keras-team/keras-tuner

PyPi (📥 390K / month · 📦 120 · ⏱️ 04.03.2024):

	pip install keras-tuner

Conda (📥 50K · ⏱️ 25.12.2024):

	conda install -c conda-forge keras-tuner

mljar-supervised (🥈29 · ⭐ 3.1K) - Python package for AutoML on Tabular Data with Feature.. MIT

GitHub (👨‍💻 29 · 🔀 410 · 📦 140 · 📋 670 - 20% open · ⏱️ 14.01.2025):

	git clone https://github.com/mljar/mljar-supervised

PyPi (📥 9.5K / month · 📦 6 · ⏱️ 14.01.2025):

	pip install mljar-supervised

Conda (📥 36K · ⏱️ 14.01.2025):

	conda install -c conda-forge mljar-supervised

lazypredict (🥈27 · ⭐ 3.1K) - Lazy Predict help build a lot of basic models without much code.. MIT

GitHub (👨‍💻 18 · 🔀 350 · 📦 1.2K · 📋 140 - 69% open · ⏱️ 03.11.2024):

	git clone https://github.com/shankarpandala/lazypredict

PyPi (📥 18K / month · 📦 6 · ⏱️ 02.11.2024):

	pip install lazypredict

Conda (📥 4.1K · ⏱️ 16.06.2023):

	conda install -c conda-forge lazypredict

Talos (🥈25 · ⭐ 1.6K · 💤) - Hyperparameter Experiments with TensorFlow and Keras. MIT

GitHub (👨‍💻 23 · 🔀 270 · 📦 200 · 📋 400 - 2% open · ⏱️ 22.04.2024):

	git clone https://github.com/autonomio/talos

PyPi (📥 1.4K / month · 📦 8 · ⏱️ 21.04.2024):

	pip install talos

FEDOT (🥈24 · ⭐ 660) - Automated modeling and machine learning framework FEDOT. BSD-3

GitHub (👨‍💻 37 · 🔀 88 · 📦 57 · 📋 560 - 11% open · ⏱️ 09.02.2025):

	git clone https://github.com/nccr-itmo/FEDOT

PyPi (📥 2.1K / month · 📦 5 · ⏱️ 28.08.2024):

	pip install fedot

featurewiz (🥉23 · ⭐ 600) - Use advanced feature engineering strategies and select best.. Apache-2

GitHub (👨‍💻 18 · 🔀 91 · 📦 81 · 📋 120 - 10% open · ⏱️ 29.01.2025):

	git clone https://github.com/AutoViML/featurewiz

PyPi (📥 15K / month · 📦 4 · ⏱️ 29.01.2025):

	pip install featurewiz

Hyperactive (🥉23 · ⭐ 520) - An optimization and data collection toolbox for convenient and fast.. MIT

GitHub (👨‍💻 9 · 🔀 42 · 📥 300 · 📦 36 · 📋 78 - 17% open · ⏱️ 24.01.2025):

	git clone https://github.com/SimonBlanke/Hyperactive

PyPi (📥 2.5K / month · 📦 13 · ⏱️ 15.08.2024):

	pip install hyperactive

Auto ViML (🥉22 · ⭐ 530) - Automatically Build Multiple ML Models with a Single Line of Code... Apache-2

GitHub (👨‍💻 9 · 🔀 100 · 📦 28 · ⏱️ 30.01.2025):

	git clone https://github.com/AutoViML/Auto_ViML

PyPi (📥 6.3K / month · 📦 3 · ⏱️ 30.01.2025):

	pip install autoviml

AlphaPy (🥉21 · ⭐ 1.4K) - Python AutoML for Trading Systems and Sports Betting. Apache-2

GitHub (👨‍💻 5 · 🔀 230 · 📦 7 · 📋 44 - 34% open · ⏱️ 15.12.2024):

	git clone https://github.com/ScottfreeLLC/AlphaPy

PyPi (📥 1.9K / month · ⏱️ 29.08.2020):

	pip install alphapy

opytimizer (🥉18 · ⭐ 610) - Opytimizer is a Python library consisting of meta-heuristic.. Apache-2

GitHub (👨‍💻 4 · 🔀 42 · 📦 19 · ⏱️ 18.08.2024):

	git clone https://github.com/gugarosa/opytimizer

PyPi (📥 610 / month · ⏱️ 18.08.2024):

	pip install opytimizer

shap-hypetune (🥉18 · ⭐ 570 · 💤) - A python package for simultaneous Hyperparameters Tuning and.. MIT

GitHub (👨‍💻 3 · 🔀 69 · 📦 21 · 📋 36 - 11% open · ⏱️ 21.02.2024):

	git clone https://github.com/cerlymarco/shap-hypetune

PyPi (📥 2.4K / month · 📦 2 · ⏱️ 21.02.2024):

	pip install shap-hypetune

Show 32 hidden projects...

TPOT (🥈32 · ⭐ 9.8K · 💤) - A Python Automated Machine Learning tool that optimizes.. ❗️LGPL-3.0

scikit-optimize (🥈32 · ⭐ 2.8K · 💀) - Sequential model-based optimization with a.. BSD-3

NNI (🥈31 · ⭐ 14K · 💀) - An open source AutoML toolkit for automate machine learning lifecycle,.. MIT

auto-sklearn (🥈31 · ⭐ 7.7K · 💀) - Automated Machine Learning with scikit-learn. BSD-3

Hyperas (🥈27 · ⭐ 2.2K · 💀) - Keras + Hyperopt: A very simple wrapper for convenient.. MIT

SMAC3 (🥈25 · ⭐ 1.1K) - SMAC3: A Versatile Bayesian Optimization Package for.. ❗️BSD-1-Clause

GPyOpt (🥈25 · ⭐ 940 · 💀) - Gaussian Process Optimization using GPy. BSD-3

AdaNet (🥈24 · ⭐ 3.5K · 💀) - Fast and flexible AutoML with learning guarantees. Apache-2

auto_ml (🥈24 · ⭐ 1.6K · 💀) - [UNMAINTAINED] Automated machine learning for analytics & production. MIT

lightwood (🥈24 · ⭐ 460) - Lightwood is Legos for Machine Learning. ❗️GPL-3.0

Orion (🥈24 · ⭐ 290 · 💀) - Asynchronous Distributed Hyperparameter Optimization. BSD-3

HpBandSter (🥉23 · ⭐ 620 · 💀) - a distributed Hyperband implementation on Steroids. BSD-3

igel (🥉21 · ⭐ 3.1K · 💀) - a delightful machine learning tool that allows you to train, test, and.. MIT

MLBox (🥉21 · ⭐ 1.5K · 💀) - MLBox is a powerful Automated Machine Learning python library. ❗️BSD-1-Clause

Test Tube (🥉21 · ⭐ 740 · 💀) - Python library to easily log experiments and parallelize.. MIT

Neuraxle (🥉21 · ⭐ 610 · 💀) - The worlds cleanest AutoML library - Do hyperparameter tuning.. Apache-2

sklearn-deap (🥉20 · ⭐ 770 · 💀) - Use evolutionary algorithms instead of gridsearch in.. MIT

optunity (🥉20 · ⭐ 420 · 💀) - optimization routines for hyperparameter tuning. BSD-3

Dragonfly (🥉19 · ⭐ 870 · 💀) - An open source python library for scalable Bayesian optimisation. MIT

Auto Tune Models (🥉19 · ⭐ 530 · 💀) - Auto Tune Models - A multi-tenant, multi-data system for.. MIT

Sherpa (🥉19 · ⭐ 340 · 💀) - Hyperparameter optimization that enables researchers to.. ❗️GPL-3.0

Advisor (🥉18 · ⭐ 1.6K · 💀) - Open-source implementation of Google Vizier for hyper parameters.. Apache-2

Xcessiv (🥉18 · ⭐ 1.3K · 💀) - A web-based application for quick, scalable, and automated.. Apache-2

HyperparameterHunter (🥉17 · ⭐ 700 · 💀) - Easy hyperparameter optimization and automatic result.. MIT

automl-gs (🥉16 · ⭐ 1.9K · 💀) - Provide an input CSV and a target field to predict, generate a.. MIT

Parfit (🥉15 · ⭐ 200 · 💀) - A package for parallelizing the fit and flexibly scoring of.. MIT

ENAS (🥉13 · ⭐ 2.7K · 💀) - PyTorch implementation of Efficient Neural Architecture Search via.. Apache-2

Auptimizer (🥉13 · ⭐ 200 · 💀) - An automatic ML model optimization tool. ❗️GPL-3.0

Hypermax (🥉13 · ⭐ 110 · 💀) - Better, faster hyper-parameter optimization. BSD-3

model_search (🥉11 · ⭐ 3.3K · 💀) - AutoML algorithms for model architecture search at scale. Apache-2

Devol (🥉11 · ⭐ 950 · 💀) - Genetic neural architecture search with Keras. MIT

Hypertunity (🥉10 · ⭐ 140 · 💀) - A toolset for black-box hyperparameter optimisation. Apache-2

Reinforcement Learning

Libraries for building and evaluating reinforcement learning & agent-based systems.

FinRL (🥇31 · ⭐ 11K) - FinRL: Financial Reinforcement Learning. MIT

GitHub (👨‍💻 120 · 🔀 2.5K · 📦 64 · 📋 730 - 33% open · ⏱️ 06.02.2025):
```
	git clone https://github.com/AI4Finance-Foundation/FinRL
```
PyPi (📥 2.3K / month · ⏱️ 08.01.2022):
```
	pip install finrl
```

TF-Agents (🥈28 · ⭐ 2.9K) - TF-Agents: A reliable, scalable and easy to use TensorFlow.. Apache-2

GitHub (👨‍💻 150 · 🔀 720 · 📋 670 - 29% open · ⏱️ 07.02.2025):

	git clone https://github.com/tensorflow/agents

PyPi (📥 38K / month · 📦 14 · ⏱️ 14.12.2023):

	pip install tf-agents

ViZDoom (🥈28 · ⭐ 1.8K) - Reinforcement Learning environments based on the 1993 game Doom. MIT

GitHub (👨‍💻 55 · 🔀 390 · 📥 12K · 📦 300 · 📋 460 - 6% open · ⏱️ 08.09.2024):

	git clone https://github.com/mwydmuch/ViZDoom

PyPi (📥 8.5K / month · 📦 15 · ⏱️ 20.08.2024):

	pip install vizdoom

Dopamine (🥈27 · ⭐ 11K) - Dopamine is a research framework for fast prototyping of.. Apache-2

GitHub (👨‍💻 15 · 🔀 1.4K · 📦 21 · 📋 190 - 54% open · ⏱️ 04.11.2024):

	git clone https://github.com/google/dopamine

PyPi (📥 25K / month · 📦 10 · ⏱️ 31.10.2024):

	pip install dopamine-rl

Acme (🥈27 · ⭐ 3.6K) - A library of reinforcement learning components and agents. Apache-2

GitHub (👨‍💻 86 · 🔀 440 · 📦 230 · 📋 270 - 23% open · ⏱️ 14.01.2025):

	git clone https://github.com/deepmind/acme

PyPi (📥 1.7K / month · 📦 3 · ⏱️ 10.02.2022):

	pip install dm-acme

Conda (📥 11K · ⏱️ 04.01.2025):

	conda install -c conda-forge dm-acme

TensorForce (🥉26 · ⭐ 3.3K · 💤) - Tensorforce: a TensorFlow library for applied.. Apache-2

GitHub (👨‍💻 85 · 🔀 530 · 📦 460 · 📋 680 - 6% open · ⏱️ 31.07.2024):

	git clone https://github.com/tensorforce/tensorforce

PyPi (📥 950 / month · 📦 4 · ⏱️ 30.08.2021):

	pip install tensorforce

PARL (🥉25 · ⭐ 3.3K) - A high-performance distributed training framework for Reinforcement.. Apache-2

GitHub (👨‍💻 46 · 🔀 810 · 📦 130 · 📋 540 - 24% open · ⏱️ 24.01.2025):

	git clone https://github.com/PaddlePaddle/PARL

PyPi (📥 1.3K / month · 📦 1 · ⏱️ 13.05.2022):

	pip install parl

RLax (🥉24 · ⭐ 1.3K) - A library of reinforcement learning building blocks in JAX. Apache-2

GitHub (👨‍💻 21 · 🔀 88 · 📦 300 · 📋 26 - 30% open · ⏱️ 15.01.2025):

	git clone https://github.com/deepmind/rlax

PyPi (📥 23K / month · 📦 11 · ⏱️ 09.01.2023):

	pip install rlax

ReAgent (🥉22 · ⭐ 3.6K) - A platform for Reasoning systems (Reinforcement Learning,.. BSD-3

GitHub (👨‍💻 170 · 🔀 510 · 📋 160 - 53% open · ⏱️ 06.01.2025):

	git clone https://github.com/facebookresearch/ReAgent

PyPi (📥 64 / month · ⏱️ 27.05.2020):

	pip install reagent

PFRL (🥉22 · ⭐ 1.2K) - PFRL: a PyTorch-based deep reinforcement learning library. MIT

GitHub (👨‍💻 20 · 🔀 150 · 📦 120 · 📋 79 - 41% open · ⏱️ 04.08.2024):

	git clone https://github.com/pfnet/pfrl

PyPi (📥 450 / month · 📦 1 · ⏱️ 16.07.2023):

	pip install pfrl

rliable (🥉13 · ⭐ 800) - [NeurIPS21 Outstanding Paper] Library for reliable evaluation on RL.. Apache-2

GitHub (👨‍💻 9 · 🔀 48 · 📦 180 · 📋 20 - 15% open · ⏱️ 12.08.2024):

	git clone https://github.com/google-research/rliable

PyPi:

	pip install rliable`

Show 12 hidden projects...

OpenAI Gym (🥇39 · ⭐ 35K · 💀) - A toolkit for developing and comparing reinforcement learning.. MIT

baselines (🥇29 · ⭐ 16K · 💀) - OpenAI Baselines: high-quality implementations of reinforcement.. MIT

keras-rl (🥈28 · ⭐ 5.5K · 💀) - Deep Reinforcement Learning for Keras. MIT

TensorLayer (🥈27 · ⭐ 7.3K · 💀) - Deep Learning and Reinforcement Learning Library for.. Apache-2

garage (🥉25 · ⭐ 1.9K · 💀) - A toolkit for reproducible reinforcement learning research. MIT

Stable Baselines (🥉24 · ⭐ 4.2K · 💀) - A fork of OpenAI Baselines, implementations of.. MIT

ChainerRL (🥉24 · ⭐ 1.2K · 💀) - ChainerRL is a deep reinforcement learning library built on top of.. MIT

TRFL (🥉22 · ⭐ 3.1K · 💀) - TensorFlow Reinforcement Learning. Apache-2

Coach (🥉21 · ⭐ 2.3K · 💀) - Reinforcement Learning Coach by Intel AI Lab enables easy.. Apache-2

SerpentAI (🥉19 · ⭐ 6.8K · 💀) - Game Agent Framework. Helping you create AIs / Bots that learn to.. MIT

DeepMind Lab (🥉17 · ⭐ 7.2K · 💀) - A customisable 3D platform for agent-based AI research. ❗Unlicensed

Maze (🥉13 · ⭐ 270 · 💀) - Maze Applied Reinforcement Learning Framework. ❗️Custom

Recommender Systems

Libraries for building and evaluating recommendation systems.

Recommenders (🥇34 · ⭐ 20K) - Best Practices on Recommendation Systems. MIT

GitHub (👨‍💻 140 · 🔀 3.1K · 📥 680 · 📦 150 · 📋 880 - 18% open · ⏱️ 19.01.2025):
```
	git clone https://github.com/microsoft/recommenders
```
PyPi (📥 20K / month · 📦 4 · ⏱️ 24.12.2024):
```
	pip install recommenders
```

torchrec (🥇30 · ⭐ 2K) - Pytorch domain library for recommendation systems. BSD-3

GitHub (👨‍💻 320 · 🔀 470 · 📦 170 · 📋 470 - 71% open · ⏱️ 13.02.2025):

	git clone https://github.com/pytorch/torchrec

PyPi (📥 2.9K / month · ⏱️ 12.05.2022):

	pip install torchrec-nightly-cpu

Cornac (🥈28 · ⭐ 920) - A Comparative Framework for Multimodal Recommender Systems. Apache-2

GitHub (👨‍💻 22 · 🔀 150 · 📦 260 · 📋 160 - 14% open · ⏱️ 26.01.2025):

	git clone https://github.com/PreferredAI/cornac

PyPi (📥 36K / month · 📦 18 · ⏱️ 24.12.2024):

	pip install cornac

Conda (📥 760K · ⏱️ 24.12.2024):

	conda install -c conda-forge cornac

scikit-surprise (🥈27 · ⭐ 6.5K · 💤) - A Python scikit for building and analyzing recommender.. BSD-3

GitHub (👨‍💻 46 · 🔀 1K · 📦 21 · 📋 400 - 21% open · ⏱️ 14.06.2024):

	git clone https://github.com/NicolasHug/Surprise

PyPi (📥 150K / month · 📦 37 · ⏱️ 19.05.2024):

	pip install scikit-surprise

Conda (📥 460K · ⏱️ 20.05.2024):

	conda install -c conda-forge scikit-surprise

TF Ranking (🥉25 · ⭐ 2.8K · 💤) - Learning to Rank in TensorFlow. Apache-2

GitHub (👨‍💻 36 · 🔀 480 · 📋 330 - 27% open · ⏱️ 18.03.2024):

	git clone https://github.com/tensorflow/ranking

PyPi (📥 52K / month · 📦 15 · ⏱️ 18.03.2024):

	pip install tensorflow_ranking

RecBole (🥉24 · ⭐ 3.6K) - A unified, comprehensive and efficient recommendation library. MIT

GitHub (👨‍💻 74 · 🔀 630 · 📋 1K - 28% open · ⏱️ 05.09.2024):

	git clone https://github.com/RUCAIBox/RecBole

PyPi (📥 49K / month · 📦 2 · ⏱️ 31.10.2023):

	pip install recbole

Conda (📥 7.4K · ⏱️ 01.11.2023):

	conda install -c aibox recbole

TF Recommenders (🥉24 · ⭐ 1.9K) - TensorFlow Recommenders is a library for building.. Apache-2

GitHub (👨‍💻 43 · 🔀 280 · 📋 450 - 59% open · ⏱️ 16.01.2025):

	git clone https://github.com/tensorflow/recommenders

PyPi (📥 210K / month · 📦 2 · ⏱️ 03.02.2023):

	pip install tensorflow-recommenders

Show 10 hidden projects...

implicit (🥈29 · ⭐ 3.6K · 💀) - Fast Python Collaborative Filtering for Implicit Feedback Datasets. MIT

lightfm (🥈28 · ⭐ 4.8K · 💀) - A Python implementation of LightFM, a hybrid recommendation.. Apache-2

lkpy (🥈27 · ⭐ 280) - Python recommendation toolkit. MIT

tensorrec (🥉21 · ⭐ 1.3K · 💀) - A TensorFlow recommendation algorithm and framework in.. Apache-2

fastFM (🥉21 · ⭐ 1.1K · 💀) - fastFM: A Library for Factorization Machines. BSD-3

recmetrics (🥉19 · ⭐ 580 · 💀) - A library of metrics for evaluating recommender systems. MIT

Spotlight (🥉18 · ⭐ 3K · 💀) - Deep recommender models using PyTorch. MIT

Case Recommender (🥉18 · ⭐ 490 · 💀) - Case Recommender: A Flexible and Extensible Python.. MIT

OpenRec (🥉17 · ⭐ 410 · 💀) - OpenRec is an open-source and modular library for neural network-.. Apache-2

Collie (🥉17 · ⭐ 110 · 💀) - A library for preparing, training, and evaluating scalable deep.. BSD-3

Privacy Machine Learning

Libraries for encrypted and privacy-preserving machine learning using methods like federated learning & differential privacy.

PySyft (🥇37 · ⭐ 9.6K · 📈) - Perform data science on data that remains in someone elses.. Apache-2

GitHub (👨‍💻 520 · 🔀 2K · 📥 2.2K · 📦 1 · 📋 3.4K - 1% open · ⏱️ 13.02.2025):
```
	git clone https://github.com/OpenMined/PySyft
```
PyPi (📥 16K / month · 📦 4 · ⏱️ 13.02.2025):
```
	pip install syft
```

Opacus (🥈31 · ⭐ 1.8K) - Training PyTorch models with differential privacy. Apache-2

GitHub (👨‍💻 85 · 🔀 350 · 📥 140 · 📦 990 · 📋 320 - 21% open · ⏱️ 12.02.2025):

	git clone https://github.com/pytorch/opacus

PyPi (📥 100K / month · 📦 36 · ⏱️ 03.08.2024):

	pip install opacus

Conda (📥 21K · ⏱️ 02.01.2025):

	conda install -c conda-forge opacus

TensorFlow Privacy (🥈25 · ⭐ 1.9K) - Library for training machine learning models with.. Apache-2

GitHub (👨‍💻 60 · 🔀 440 · 📥 190 · 📋 210 - 55% open · ⏱️ 10.02.2025):

	git clone https://github.com/tensorflow/privacy

PyPi (📥 19K / month · 📦 21 · ⏱️ 14.02.2024):

	pip install tensorflow-privacy

TFEncrypted (🥈25 · ⭐ 1.2K) - A Framework for Encrypted Machine Learning in TensorFlow. Apache-2

GitHub (👨‍💻 29 · 🔀 210 · 📦 68 · 📋 440 - 32% open · ⏱️ 25.09.2024):

	git clone https://github.com/tf-encrypted/tf-encrypted

PyPi (📥 1.3K / month · 📦 9 · ⏱️ 16.11.2022):

	pip install tf-encrypted

FATE (🥉23 · ⭐ 5.8K) - An Industrial Grade Federated Learning Framework. Apache-2

GitHub (👨‍💻 100 · 🔀 1.6K · 📋 2.1K - 3% open · ⏱️ 19.11.2024):

	git clone https://github.com/FederatedAI/FATE

PyPi (⏱️ 06.05.2020):

	pip install ETAF

CrypTen (🥉23 · ⭐ 1.6K) - A framework for Privacy Preserving Machine Learning. MIT

GitHub (👨‍💻 39 · 🔀 280 · 📦 53 · 📋 280 - 28% open · ⏱️ 23.11.2024):

	git clone https://github.com/facebookresearch/CrypTen

PyPi (📥 510 / month · 📦 1 · ⏱️ 08.12.2022):

	pip install crypten

Show 1 hidden projects...

PipelineDP (🥉19 · ⭐ 280) - PipelineDP is a Python framework for applying differentially.. Apache-2

Workflow & Experiment Tracking

Libraries to organize, track, and visualize machine learning experiments.

mlflow (🥇44 · ⭐ 19K) - Open source platform for the machine learning lifecycle. Apache-2

GitHub (👨‍💻 820 · 🔀 4.3K · 📦 52K · 📋 4.4K - 38% open · ⏱️ 13.02.2025):
```
	git clone https://github.com/mlflow/mlflow
```
PyPi (📥 14M / month · 📦 990 · ⏱️ 13.02.2025):
```
	pip install mlflow
```
Conda (📥 3M · ⏱️ 31.01.2025):
```
	conda install -c conda-forge mlflow
```

wandb client (🥇42 · ⭐ 9.5K) - The AI developer platform. Use Weights & Biases to train and fine-.. MIT

GitHub (👨‍💻 210 · 🔀 700 · 📥 610 · 📦 66K · 📋 3.5K - 17% open · ⏱️ 13.02.2025):

	git clone https://github.com/wandb/client

PyPi (📥 17M / month · 📦 1.6K · ⏱️ 05.02.2025):

	pip install wandb

Conda (📥 950K · ⏱️ 06.02.2025):

	conda install -c conda-forge wandb

Tensorboard (🥇42 · ⭐ 6.8K) - TensorFlows Visualization Toolkit. Apache-2

GitHub (👨‍💻 320 · 🔀 1.7K · 📦 290K · 📋 1.9K - 35% open · ⏱️ 30.01.2025):

	git clone https://github.com/tensorflow/tensorboard

PyPi (📥 24M / month · 📦 2.5K · ⏱️ 12.02.2025):

	pip install tensorboard

Conda (📥 5.3M · ⏱️ 10.12.2024):

	conda install -c conda-forge tensorboard

DVC (🥇41 · ⭐ 14K) - Data Versioning and ML Experiments. Apache-2

GitHub (👨‍💻 310 · 🔀 1.2K · 📦 21K · 📋 4.7K - 5% open · ⏱️ 09.02.2025):

	git clone https://github.com/iterative/dvc

PyPi (📥 820K / month · 📦 140 · ⏱️ 13.01.2025):

	pip install dvc

Conda (📥 2.6M · ⏱️ 13.01.2025):

	conda install -c conda-forge dvc

SageMaker SDK (🥇41 · ⭐ 2.1K) - A library for training and deploying machine learning.. Apache-2

GitHub (👨‍💻 470 · 🔀 1.1K · 📦 5.3K · 📋 1.6K - 22% open · ⏱️ 13.02.2025):

	git clone https://github.com/aws/sagemaker-python-sdk

PyPi (📥 24M / month · 📦 160 · ⏱️ 01.02.2025):

	pip install sagemaker

Conda (📥 1.4M · ⏱️ 01.02.2025):

	conda install -c conda-forge sagemaker-python-sdk

PyCaret (🥈35 · ⭐ 9.1K) - An open-source, low-code machine learning library in Python. MIT

GitHub (👨‍💻 140 · 🔀 1.8K · 📥 720 · 📦 7.1K · 📋 2.3K - 16% open · ⏱️ 11.02.2025):

	git clone https://github.com/pycaret/pycaret

PyPi (📥 300K / month · 📦 31 · ⏱️ 28.04.2024):

	pip install pycaret

Conda (📥 64K · ⏱️ 20.01.2025):

	conda install -c conda-forge pycaret

Metaflow (🥈35 · ⭐ 8.5K) - Open Source AI/ML Platform. Apache-2

GitHub (👨‍💻 100 · 🔀 790 · 📦 800 · 📋 760 - 42% open · ⏱️ 12.02.2025):

	git clone https://github.com/Netflix/metaflow

PyPi (📥 300K / month · 📦 50 · ⏱️ 11.02.2025):

	pip install metaflow

Conda (📥 270K · ⏱️ 05.02.2025):

	conda install -c conda-forge metaflow

tensorboardX (🥈34 · ⭐ 7.9K) - tensorboard for pytorch (and chainer, mxnet, numpy, ...). MIT

GitHub (👨‍💻 82 · 🔀 860 · 📥 460 · 📦 53K · 📋 460 - 17% open · ⏱️ 01.01.2025):

	git clone https://github.com/lanpa/tensorboardX

PyPi (📥 2.6M / month · 📦 620 · ⏱️ 20.08.2023):

	pip install tensorboardX

Conda (📥 1.3M · ⏱️ 20.12.2024):

	conda install -c conda-forge tensorboardx

ClearML (🥈33 · ⭐ 5.8K) - ClearML - Auto-Magical CI/CD to streamline your AI workload... Apache-2

GitHub (👨‍💻 100 · 🔀 660 · 📥 3.1K · 📦 1.6K · 📋 1.1K - 46% open · ⏱️ 12.02.2025):

	git clone https://github.com/allegroai/clearml

PyPi (📥 390K / month · 📦 53 · ⏱️ 05.02.2025):

	pip install clearml

Docker Hub (📥 30K · ⏱️ 05.10.2020):

	docker pull allegroai/trains

snakemake (🥈33 · ⭐ 2.4K · 📉) - This is the development home of the workflow management system.. MIT

GitHub (👨‍💻 350 · 🔀 560 · 📦 2.2K · 📋 1.9K - 61% open · ⏱️ 12.02.2025):

	git clone https://github.com/snakemake/snakemake

PyPi (📥 73K / month · 📦 260 · ⏱️ 12.02.2025):

	pip install snakemake

Conda (📥 1.3M · ⏱️ 12.02.2025):

	conda install -c bioconda snakemake

kaggle (🥈32 · ⭐ 6.4K) - Official Kaggle API. Apache-2

GitHub (👨‍💻 49 · 🔀 1.1K · 📦 21 · 📋 500 - 30% open · ⏱️ 10.02.2025):

	git clone https://github.com/Kaggle/kaggle-api

PyPi (📥 250K / month · 📦 230 · ⏱️ 06.02.2025):

	pip install kaggle

Conda (📥 210K · ⏱️ 13.01.2025):

	conda install -c conda-forge kaggle

aim (🥈32 · ⭐ 5.4K) - Aim An easy-to-use & supercharged open-source experiment tracker. Apache-2

GitHub (👨‍💻 80 · 🔀 330 · 📦 810 · 📋 1.1K - 36% open · ⏱️ 13.02.2025):

	git clone https://github.com/aimhubio/aim

PyPi (📥 180K / month · 📦 41 · ⏱️ 12.02.2025):

	pip install aim

Conda (📥 110K · ⏱️ 14.06.2024):

	conda install -c conda-forge aim

AzureML SDK (🥈32 · ⭐ 4.1K) - Python notebooks with ML and deep learning examples with Azure.. MIT

GitHub (👨‍💻 64 · 🔀 2.5K · 📥 660 · 📋 1.5K - 26% open · ⏱️ 16.12.2024):

	git clone https://github.com/Azure/MachineLearningNotebooks

PyPi (📥 400K / month · 📦 31 · ⏱️ 10.12.2024):

	pip install azureml-sdk

VisualDL (🥈29 · ⭐ 4.8K) - Deep Learning Visualization Toolkit. Apache-2

GitHub (👨‍💻 36 · 🔀 630 · 📥 500 · 📦 2 · 📋 500 - 28% open · ⏱️ 22.01.2025):

	git clone https://github.com/PaddlePaddle/VisualDL

PyPi (📥 140K / month · 📦 82 · ⏱️ 30.10.2024):

	pip install visualdl

sacred (🥈29 · ⭐ 4.3K · 📈) - Sacred is a tool to help you configure, organize, log and reproduce.. MIT

GitHub (👨‍💻 110 · 🔀 380 · 📦 3.4K · 📋 560 - 18% open · ⏱️ 26.11.2024):

	git clone https://github.com/IDSIA/sacred

PyPi (📥 23K / month · 📦 60 · ⏱️ 26.11.2024):

	pip install sacred

Conda (📥 7.9K · ⏱️ 28.11.2023):

	conda install -c conda-forge sacred

Neptune.ai (🥈29 · ⭐ 600) - The experiment tracker for foundation model training. Apache-2

GitHub (👨‍💻 55 · 🔀 64 · 📦 740 · 📋 250 - 13% open · ⏱️ 19.12.2024):

	git clone https://github.com/neptune-ai/neptune-client

PyPi (📥 510K / month · 📦 77 · ⏱️ 31.10.2024):

	pip install neptune-client

Conda (📥 320K · ⏱️ 31.10.2024):

	conda install -c conda-forge neptune-client

livelossplot (🥉28 · ⭐ 1.3K) - Live training loss plot in Jupyter Notebook for Keras,.. MIT

GitHub (👨‍💻 17 · 🔀 140 · 📦 1.8K · 📋 79 - 7% open · ⏱️ 03.01.2025):

	git clone https://github.com/stared/livelossplot

PyPi (📥 22K / month · 📦 16 · ⏱️ 03.01.2025):

	pip install livelossplot

Labml (🥉26 · ⭐ 2.1K) - Monitor deep learning model training and hardware usage from your mobile.. MIT

GitHub (👨‍💻 9 · 🔀 140 · 📦 200 · 📋 50 - 12% open · ⏱️ 30.01.2025):

	git clone https://github.com/labmlai/labml

PyPi (📥 5.4K / month · 📦 14 · ⏱️ 15.09.2024):

	pip install labml

ml-metadata (🥉26 · ⭐ 640) - For recording and retrieving metadata associated with ML.. Apache-2

GitHub (👨‍💻 21 · 🔀 150 · 📥 3K · 📦 640 · 📋 120 - 38% open · ⏱️ 24.10.2024):

	git clone https://github.com/google/ml-metadata

PyPi (📥 180K / month · 📦 31 · ⏱️ 01.10.2024):

	pip install ml-metadata

TNT (🥉25 · ⭐ 1.7K) - A lightweight library for PyTorch training tools and utilities. BSD-3

GitHub (👨‍💻 140 · 🔀 280 · 📋 150 - 56% open · ⏱️ 03.02.2025):

	git clone https://github.com/pytorch/tnt

PyPi (📥 4.8K / month · 📦 24 · ⏱️ 29.07.2018):

	pip install torchnet

gokart (🥉25 · ⭐ 320) - Gokart solves reproducibility, task dependencies, constraints of good code,.. MIT

GitHub (👨‍💻 44 · 🔀 61 · 📦 84 · 📋 88 - 26% open · ⏱️ 13.02.2025):

	git clone https://github.com/m3dev/gokart

PyPi (📥 5.7K / month · 📦 8 · ⏱️ 13.01.2025):

	pip install gokart

quinn (🥉24 · ⭐ 660) - pyspark methods to enhance developer productivity. Apache-2

GitHub (👨‍💻 31 · 🔀 98 · 📥 56 · 📦 90 · 📋 130 - 27% open · ⏱️ 06.12.2024):

	git clone https://github.com/MrPowers/quinn

PyPi (📥 740K / month · 📦 7 · ⏱️ 13.02.2024):

	pip install quinn

keepsake (🥉18 · ⭐ 1.7K) - Version control for machine learning. Apache-2

GitHub (👨‍💻 18 · 🔀 71 · 📋 190 - 66% open · ⏱️ 03.12.2024):

	git clone https://github.com/replicate/keepsake

PyPi (📥 330 / month · 📦 1 · ⏱️ 25.01.2021):

	pip install keepsake

CometML (🥉15) - Supercharging Machine Learning. MIT

GitHub:

	git clone https://github.com/comet-ml/examples

PyPi (📥 210K / month · 📦 88 · ⏱️ 03.02.2025):

	pip install comet_ml

Conda:

	conda install -c anaconda comet_ml

Show 16 hidden projects...

Catalyst (🥉28 · ⭐ 3.3K · 💀) - Accelerated deep learning R&D. Apache-2

knockknock (🥉27 · ⭐ 2.8K · 💀) - Knock Knock: Get notified when your training ends with only two.. MIT

Guild AI (🥉23 · ⭐ 880 · 💀) - Experiment tracking, ML developer tools. Apache-2

SKLL (🥉23 · ⭐ 550) - SciKit-Learn Laboratory (SKLL) makes it easy to run machine.. ❗Unlicensed

hiddenlayer (🥉22 · ⭐ 1.8K · 💀) - Neural network graphs and training metrics for.. MIT

Studio.ml (🥉22 · ⭐ 380 · 💀) - Studio: Simplify and expedite model building process. Apache-2

lore (🥉21 · ⭐ 1.6K · 💀) - Lore makes machine learning approachable for Software Engineers and.. MIT

TensorBoard Logger (🥉21 · ⭐ 630 · 💀) - Log TensorBoard events without touching TensorFlow. MIT

TensorWatch (🥉20 · ⭐ 3.4K · 💀) - Debugging, monitoring and visualization for Python Machine.. MIT

MXBoard (🥉20 · ⭐ 320 · 💀) - Logging MXNet data for visualization in TensorBoard. Apache-2

datmo (🥉18 · ⭐ 340 · 💀) - Open source production model management tool for data scientists. MIT

chitra (🥉17 · ⭐ 220 · 💤) - A multi-functional library for full-stack Deep Learning… Apache-2

steppy (🥉17 · ⭐ 130 · 💀) - Lightweight, Python library for fast and reproducible experimentation. MIT

caliban (🥉16 · ⭐ 500 · 💀) - Research workflows made easy, locally and in the Cloud. Apache-2

ModelChimp (🥉13 · ⭐ 130 · 💀) - Experiment tracking for machine and deep learning projects. BSD-2

traintool (🥉9 · ⭐ 12 · 💀) - Train off-the-shelf machine learning models in one.. Apache-2

Model Serialization & Deployment

Libraries to serialize models to files, convert between a variety of model formats, and optimize models for deployment.

onnx (🥇43 · ⭐ 18K) - Open standard for machine learning interoperability. Apache-2

GitHub (👨‍💻 330 · 🔀 3.7K · 📥 23K · 📦 40K · 📋 2.9K - 11% open · ⏱️ 12.02.2025):
```
	git clone https://github.com/onnx/onnx
```
PyPi (📥 5.8M / month · 📦 1.1K · ⏱️ 01.10.2024):
```
	pip install onnx
```
Conda (📥 1.6M · ⏱️ 31.12.2024):
```
	conda install -c conda-forge onnx
```

triton (🥇43 · ⭐ 14K) - Development repository for the Triton language and compiler. MIT

GitHub (👨‍💻 370 · 🔀 1.7K · 📦 53K · 📋 1.6K - 42% open · ⏱️ 13.02.2025):

	git clone https://github.com/openai/triton

PyPi (📥 19M / month · 📦 340 · ⏱️ 22.01.2025):

	pip install triton

huggingface_hub (🥈38 · ⭐ 2.3K) - The official Python client for the Huggingface Hub. Apache-2

GitHub (👨‍💻 220 · 🔀 610 · 📋 1.1K - 14% open · ⏱️ 13.02.2025):

	git clone https://github.com/huggingface/huggingface_hub

PyPi (📥 58M / month · 📦 2.4K · ⏱️ 30.01.2025):

	pip install huggingface_hub

Conda (📥 2.7M · ⏱️ 31.01.2025):

	conda install -c conda-forge huggingface_hub

Core ML Tools (🥈36 · ⭐ 4.6K) - Core ML tools contain supporting tools for Core ML model.. BSD-3

GitHub (👨‍💻 180 · 🔀 660 · 📥 13K · 📦 4.5K · 📋 1.5K - 24% open · ⏱️ 11.02.2025):

	git clone https://github.com/apple/coremltools

PyPi (📥 700K / month · 📦 87 · ⏱️ 21.01.2025):

	pip install coremltools

Conda (📥 88K · ⏱️ 27.12.2024):

	conda install -c conda-forge coremltools

BentoML (🥈35 · ⭐ 7.4K) - The easiest way to serve AI apps and models - Build Model Inference.. Apache-2

GitHub (👨‍💻 220 · 🔀 800 · 📥 600 · 📦 2.4K · 📋 1.1K - 14% open · ⏱️ 13.02.2025):

	git clone https://github.com/bentoml/BentoML

PyPi (📥 100K / month · 📦 34 · ⏱️ 13.02.2025):

	pip install bentoml

TorchServe (🥈33 · ⭐ 4.3K) - Serve, optimize and scale PyTorch models in production. Apache-2

GitHub (👨‍💻 220 · 🔀 870 · 📥 7.6K · 📦 810 · 📋 1.7K - 25% open · ⏱️ 20.12.2024):

	git clone https://github.com/pytorch/serve

PyPi (📥 52K / month · 📦 22 · ⏱️ 30.09.2024):

	pip install torchserve

Conda (📥 460K · ⏱️ 30.09.2024):

	conda install -c pytorch torchserve

Docker Hub (📥 1.4M · ⭐ 31 · ⏱️ 30.09.2024):

	docker pull pytorch/torchserve

hls4ml (🥈29 · ⭐ 1.4K) - Machine learning on FPGAs using HLS. Apache-2

GitHub (👨‍💻 63 · 🔀 420 · 📦 39 · 📋 460 - 41% open · ⏱️ 08.02.2025):

	git clone https://github.com/fastmachinelearning/hls4ml

PyPi (📥 1.7K / month · 📦 1 · ⏱️ 09.12.2024):

	pip install hls4ml

Conda (📥 9.8K · ⏱️ 16.06.2023):

	conda install -c conda-forge hls4ml

Hummingbird (🥈25 · ⭐ 3.4K) - Hummingbird compiles trained ML models into tensor computation for.. MIT

GitHub (👨‍💻 40 · 🔀 280 · 📥 790 · 📋 330 - 20% open · ⏱️ 24.10.2024):

	git clone https://github.com/microsoft/hummingbird

PyPi (📥 6.6K / month · 📦 7 · ⏱️ 25.10.2024):

	pip install hummingbird-ml

Conda (📥 55K · ⏱️ 20.11.2024):

	conda install -c conda-forge hummingbird-ml

nebullvm (🥉21 · ⭐ 8.4K · 💤) - A collection of libraries to optimise AI model performances. Apache-2

GitHub (👨‍💻 40 · 🔀 630 · 📋 200 - 49% open · ⏱️ 22.07.2024):

	git clone https://github.com/nebuly-ai/nebullvm

PyPi (📥 810 / month · 📦 2 · ⏱️ 18.06.2023):

	pip install nebullvm

tfdeploy (🥉17 · ⭐ 350) - Deploy tensorflow graphs for fast evaluation and export to.. BSD-3

GitHub (👨‍💻 4 · 🔀 38 · 📋 34 - 32% open · ⏱️ 04.01.2025):

	git clone https://github.com/riga/tfdeploy

PyPi (📥 320 / month · ⏱️ 30.03.2017):

	pip install tfdeploy

Show 10 hidden projects...

mmdnn (🥈25 · ⭐ 5.8K · 💀) - MMdnn is a set of tools to help users inter-operate among different deep.. MIT

m2cgen (🥈25 · ⭐ 2.8K · 💀) - Transform ML models into a native code (Java, C, Python, Go,.. MIT

sklearn-porter (🥉24 · ⭐ 1.3K · 💀) - Transpile trained scikit-learn estimators to C, Java,.. BSD-3

cortex (🥉22 · ⭐ 8K · 💀) - Production infrastructure for machine learning at scale. Apache-2

OMLT (🥉20 · ⭐ 300) - Represent trained machine learning models as Pyomo optimization.. ❗Unlicensed

Larq Compute Engine (🥉20 · ⭐ 250) - Highly optimized inference engine for Binarized.. Apache-2

pytorch2keras (🥉19 · ⭐ 860 · 💀) - PyTorch to Keras model convertor. MIT

modelkit (🥉17 · ⭐ 150 · 💤) - Toolkit for developing and maintaining ML models. MIT

backprop (🥉15 · ⭐ 240 · 💀) - Backprop makes it simple to use, finetune, and deploy state-of-.. Apache-2

ml-ane-transformers (🥉13 · ⭐ 2.6K · 💀) - Reference implementation of the Transformer.. ❗Unlicensed

Model Interpretability

Libraries to visualize, explain, debug, evaluate, and interpret machine learning models.

shap (🥇41 · ⭐ 23K) - A game theoretic approach to explain the output of any machine learning model. MIT

GitHub (👨‍💻 260 · 🔀 3.3K · 📦 25K · 📋 2.6K - 26% open · ⏱️ 11.02.2025):
```
	git clone https://github.com/slundberg/shap
```
PyPi (📥 8.6M / month · 📦 750 · ⏱️ 27.06.2024):
```
	pip install shap
```
Conda (📥 5.1M · ⏱️ 10.12.2024):
```
	conda install -c conda-forge shap
```

Netron (🥇37 · ⭐ 29K) - Visualizer for neural network, deep learning and machine learning.. MIT

GitHub (👨‍💻 2 · 🔀 2.8K · 📥 70K · 📦 650 · 📋 1.2K - 1% open · ⏱️ 13.02.2025):

	git clone https://github.com/lutzroeder/netron

PyPi (📥 27K / month · 📦 83 · ⏱️ 08.02.2025):

	pip install netron

arviz (🥇36 · ⭐ 1.6K) - Exploratory analysis of Bayesian models with Python. Apache-2

GitHub (👨‍💻 170 · 🔀 420 · 📥 180 · 📦 9.1K · 📋 880 - 20% open · ⏱️ 07.02.2025):

	git clone https://github.com/arviz-devs/arviz

PyPi (📥 1.7M / month · 📦 310 · ⏱️ 28.09.2024):

	pip install arviz

Conda (📥 2.3M · ⏱️ 13.12.2024):

	conda install -c conda-forge arviz

InterpretML (🥇34 · ⭐ 6.4K) - Fit interpretable models. Explain blackbox machine learning. MIT

GitHub (👨‍💻 48 · 🔀 740 · 📦 830 · 📋 460 - 22% open · ⏱️ 11.02.2025):

	git clone https://github.com/interpretml/interpret

PyPi (📥 160K / month · 📦 50 · ⏱️ 06.01.2025):

	pip install interpret

Captum (🥇33 · ⭐ 5.1K · 📉) - Model interpretability and understanding for PyTorch. BSD-3

GitHub (👨‍💻 120 · 🔀 500 · 📦 2.8K · 📋 590 - 42% open · ⏱️ 22.01.2025):

	git clone https://github.com/pytorch/captum

PyPi (📥 260K / month · 📦 130 · ⏱️ 05.12.2023):

	pip install captum

Conda (📥 110K · ⏱️ 14.01.2025):

	conda install -c conda-forge captum

evaluate (🥈32 · ⭐ 2.1K) - Evaluate: A library for easily evaluating machine learning models.. Apache-2

GitHub (👨‍💻 130 · 🔀 260 · 📦 17K · 📋 360 - 60% open · ⏱️ 10.01.2025):

	git clone https://github.com/huggingface/evaluate

PyPi (📥 2.5M / month · 📦 400 · ⏱️ 11.09.2024):

	pip install evaluate

shapash (🥈31 · ⭐ 2.8K) - Shapash: User-friendly Explainability and Interpretability to.. Apache-2

GitHub (👨‍💻 40 · 🔀 330 · 📦 190 · 📋 230 - 17% open · ⏱️ 13.02.2025):

	git clone https://github.com/MAIF/shapash

PyPi (📥 9.8K / month · 📦 4 · ⏱️ 13.02.2025):

	pip install shapash

explainerdashboard (🥈31 · ⭐ 2.4K) - Quickly build Explainable AI dashboards that show the inner.. MIT

GitHub (👨‍💻 21 · 🔀 330 · 📦 600 · 📋 240 - 15% open · ⏱️ 29.12.2024):

	git clone https://github.com/oegedijk/explainerdashboard

PyPi (📥 69K / month · 📦 13 · ⏱️ 29.12.2024):

	pip install explainerdashboard

Conda (📥 61K · ⏱️ 18.03.2024):

	conda install -c conda-forge explainerdashboard

fairlearn (🥈30 · ⭐ 2K) - A Python package to assess and improve fairness of machine learning.. MIT

GitHub (👨‍💻 92 · 🔀 420 · 📦 3 · 📋 510 - 31% open · ⏱️ 12.02.2025):

	git clone https://github.com/fairlearn/fairlearn

PyPi (📥 140K / month · 📦 63 · ⏱️ 11.12.2024):

	pip install fairlearn

Conda (📥 42K · ⏱️ 02.01.2025):

	conda install -c conda-forge fairlearn

pyLDAvis (🥈30 · ⭐ 1.8K · 💤) - Python library for interactive topic model visualization... BSD-3

GitHub (👨‍💻 42 · 🔀 360 · 📦 6.9K · 📋 190 - 40% open · ⏱️ 29.04.2024):

	git clone https://github.com/bmabey/pyLDAvis

PyPi (📥 130K / month · 📦 100 · ⏱️ 23.04.2023):

	pip install pyldavis

Conda (📥 92K · ⏱️ 14.01.2025):

	conda install -c conda-forge pyldavis

DoWhy (🥈29 · ⭐ 7.3K) - DoWhy is a Python library for causal inference that supports explicit.. MIT

GitHub (👨‍💻 100 · 🔀 940 · 📥 42 · 📦 520 · 📋 490 - 27% open · ⏱️ 21.01.2025):

	git clone https://github.com/py-why/dowhy

PyPi (📥 43K / month · 📦 18 · ⏱️ 24.11.2024):

	pip install dowhy

Conda (📥 38K · ⏱️ 25.11.2024):

	conda install -c conda-forge dowhy

dtreeviz (🥈28 · ⭐ 3K) - A python library for decision tree visualization and model interpretation. MIT

GitHub (👨‍💻 27 · 🔀 340 · 📦 1.4K · 📋 210 - 34% open · ⏱️ 29.08.2024):

	git clone https://github.com/parrt/dtreeviz

PyPi (📥 120K / month · 📦 53 · ⏱️ 07.07.2022):

	pip install dtreeviz

Conda (📥 98K · ⏱️ 13.07.2023):

	conda install -c conda-forge dtreeviz

Model Analysis (🥈28 · ⭐ 1.3K) - Model analysis tools for TensorFlow. Apache-2

GitHub (👨‍💻 59 · 🔀 280 · 📋 97 - 39% open · ⏱️ 08.01.2025):

	git clone https://github.com/tensorflow/model-analysis

PyPi (📥 92K / month · 📦 19 · ⏱️ 05.12.2024):

	pip install tensorflow-model-analysis

LIT (🥈27 · ⭐ 3.5K) - The Learning Interpretability Tool: Interactively analyze ML models to.. Apache-2

GitHub (👨‍💻 38 · 🔀 360 · 📦 42 · 📋 210 - 57% open · ⏱️ 20.12.2024):

	git clone https://github.com/PAIR-code/lit

PyPi (📥 3.1K / month · 📦 3 · ⏱️ 20.12.2024):

	pip install lit-nlp

Conda (📥 110K · ⏱️ 16.06.2023):

	conda install -c conda-forge lit-nlp

Fairness 360 (🥈26 · ⭐ 2.5K) - A comprehensive set of fairness metrics for datasets and.. Apache-2

GitHub (👨‍💻 73 · 🔀 840 · 📦 570 · 📋 300 - 65% open · ⏱️ 10.12.2024):

	git clone https://github.com/Trusted-AI/AIF360

PyPi (📥 20K / month · 📦 32 · ⏱️ 08.04.2024):

	pip install aif360

Conda (📥 19K · ⏱️ 05.01.2025):

	conda install -c conda-forge aif360

responsible-ai-widgets (🥈26 · ⭐ 1.5K) - Responsible AI Toolbox is a suite of tools providing.. MIT

GitHub (👨‍💻 43 · 🔀 380 · 📋 320 - 26% open · ⏱️ 07.02.2025):

	git clone https://github.com/microsoft/responsible-ai-toolbox

PyPi (📥 8.8K / month · 📦 6 · ⏱️ 08.07.2024):

	pip install raiwidgets

imodels (🥈26 · ⭐ 1.4K) - Interpretable ML package for concise, transparent, and accurate.. MIT

GitHub (👨‍💻 24 · 🔀 120 · 📦 110 · 📋 95 - 38% open · ⏱️ 08.01.2025):

	git clone https://github.com/csinva/imodels

PyPi (📥 26K / month · 📦 9 · ⏱️ 15.10.2024):

	pip install imodels

CausalNex (🥉24 · ⭐ 2.3K · 💤) - A Python library that helps data scientists to infer.. Apache-2

GitHub (👨‍💻 40 · 🔀 260 · 📦 140 · 📋 140 - 17% open · ⏱️ 10.02.2024):

	git clone https://github.com/quantumblacklabs/causalnex

PyPi (📥 2.7K / month · 📦 4 · ⏱️ 22.06.2023):

	pip install causalnex

Explainability 360 (🥉23 · ⭐ 1.7K · 💤) - Interpretability and explainability of data and.. Apache-2

GitHub (👨‍💻 41 · 🔀 300 · 📦 110 · 📋 85 - 63% open · ⏱️ 16.07.2024):

	git clone https://github.com/Trusted-AI/AIX360

PyPi (📥 960 / month · 📦 1 · ⏱️ 31.07.2023):

	pip install aix360

aequitas (🥉23 · ⭐ 710) - Bias Auditing & Fair ML Toolkit. MIT

GitHub (👨‍💻 22 · 🔀 120 · 📦 180 · 📋 99 - 51% open · ⏱️ 11.09.2024):

	git clone https://github.com/dssg/aequitas

PyPi (📥 24K / month · 📦 8 · ⏱️ 30.01.2024):

	pip install aequitas

What-If Tool (🥉22 · ⭐ 930 · 💤) - Source code/webpage/demos for the What-If Tool. Apache-2

GitHub (👨‍💻 20 · 🔀 170 · 📦 2 · 📋 150 - 61% open · ⏱️ 01.02.2024):

	git clone https://github.com/PAIR-code/what-if-tool

PyPi (📥 8.2K / month · 📦 6 · ⏱️ 12.10.2021):

	pip install witwidget

Conda (📥 2.4M · ⏱️ 16.06.2023):

	conda install -c conda-forge tensorboard-plugin-wit

npm (📥 430 / month · 📦 3 · ⏱️ 12.10.2021):

	npm install wit-widget

random-forest-importances (🥉22 · ⭐ 610) - Code to compute permutation and drop-column.. MIT

GitHub (👨‍💻 15 · 🔀 130 · 📦 180 · 📋 39 - 20% open · ⏱️ 29.09.2024):

	git clone https://github.com/parrt/random-forest-importances

PyPi (📥 12K / month · 📦 5 · ⏱️ 28.01.2021):

	pip install rfpimp

ecco (🥉21 · ⭐ 2K) - Explain, analyze, and visualize NLP language models. Ecco creates.. BSD-3

GitHub (👨‍💻 12 · 🔀 160 · 📥 140 · 📦 32 · 📋 64 - 51% open · ⏱️ 15.08.2024):

	git clone https://github.com/jalammar/ecco

PyPi (📥 860 / month · 📦 1 · ⏱️ 09.01.2022):

	pip install ecco

Conda (📥 6.2K · ⏱️ 16.06.2023):

	conda install -c conda-forge ecco

fairness-indicators (🥉21 · ⭐ 350) - Tensorflows Fairness Evaluation and Visualization.. Apache-2

GitHub (👨‍💻 36 · 🔀 80 · 📋 39 - 74% open · ⏱️ 22.01.2025):

	git clone https://github.com/tensorflow/fairness-indicators

PyPi (📥 1.7K / month · ⏱️ 22.01.2025):

	pip install fairness-indicators

DiCE (🥉20 · ⭐ 1.4K) - Generate Diverse Counterfactual Explanations for any machine.. MIT

GitHub (👨‍💻 19 · 🔀 190 · 📋 180 - 48% open · ⏱️ 22.11.2024):

	git clone https://github.com/interpretml/DiCE

PyPi (📥 32K / month · 📦 6 · ⏱️ 27.10.2023):

	pip install dice-ml

ExplainX.ai (🥉16 · ⭐ 420) - Explainable AI framework for data scientists. Explain & debug any.. MIT

GitHub (👨‍💻 5 · 🔀 53 · 📥 20 · 📋 39 - 25% open · ⏱️ 21.08.2024):

	git clone https://github.com/explainX/explainx

PyPi (📥 1.2K / month · ⏱️ 04.02.2021):

	pip install explainx

interpret-text (🥉15 · ⭐ 420 · 💤) - A library that incorporates state-of-the-art explainers.. MIT

GitHub (👨‍💻 18 · 🔀 69 · 📋 100 - 84% open · ⏱️ 05.02.2024):

	git clone https://github.com/interpretml/interpret-text

PyPi (📥 160 / month · ⏱️ 07.12.2021):

	pip install interpret-text

Show 27 hidden projects...

Lime (🥇33 · ⭐ 12K · 💀) - Lime: Explaining the predictions of any machine learning classifier. BSD-2

Deep Checks (🥈29 · ⭐ 3.7K) - Deepchecks: Tests for Continuous Validation of ML Models &.. ❗️AGPL-3.0

yellowbrick (🥈27 · ⭐ 4.3K · 💀) - Visual analysis and diagnostic tools to facilitate.. Apache-2

Alibi (🥈27 · ⭐ 2.4K) - Algorithms for explaining machine learning models. ❗️Intel

scikit-plot (🥈27 · ⭐ 2.4K · 💀) - An intuitive library to add plotting functionality to.. MIT

DALEX (🥈27 · ⭐ 1.4K) - moDel Agnostic Language for Exploration and eXplanation. ❗️GPL-3.0

eli5 (🥈26 · ⭐ 2.8K · 💀) - A library for debugging/inspecting machine learning classifiers and.. MIT

Lucid (🥉25 · ⭐ 4.7K · 💀) - A collection of infrastructure and tools for research in.. Apache-2

keras-vis (🥉25 · ⭐ 3K · 💀) - Neural network visualization toolkit for keras. MIT

iNNvestigate (🥉25 · ⭐ 1.3K · 💀) - A toolbox to iNNvestigate neural networks predictions!. BSD-2

checklist (🥉24 · ⭐ 2K · 💀) - Beyond Accuracy: Behavioral Testing of NLP models with CheckList. MIT

keract (🥉23 · ⭐ 1.1K · 💀) - Layers Outputs and Gradients in Keras. Made easy. MIT

TreeInterpreter (🥉23 · ⭐ 750 · 💀) - Package for interpreting scikit-learns decision tree.. BSD-3

Quantus (🥉23 · ⭐ 580) - Quantus is an eXplainable AI toolkit for responsible evaluation of.. ❗️GPL-3.0

tf-explain (🥉22 · ⭐ 1K · 💀) - Interpretability Methods for tf.keras models with Tensorflow.. MIT

deeplift (🥉21 · ⭐ 850 · 💀) - Public facing deeplift repo. MIT

tcav (🥉20 · ⭐ 630 · 💀) - Code for the TCAV ML interpretability project. Apache-2

XAI (🥉19 · ⭐ 1.2K · 💀) - XAI - An eXplainability toolbox for machine learning. MIT

LOFO (🥉18 · ⭐ 830 · 💀) - Leave One Feature Out Importance. MIT

sklearn-evaluation (🥉18 · ⭐ 460 · 💀) - Machine learning model evaluation made easy: plots,.. MIT

model-card-toolkit (🥉18 · ⭐ 430 · 💀) - A toolkit that streamlines and automates the.. Apache-2

Anchor (🥉16 · ⭐ 800 · 💀) - Code for High-Precision Model-Agnostic Explanations paper. BSD-2

FlashTorch (🥉16 · ⭐ 740 · 💀) - Visualization toolkit for neural networks in PyTorch! Demo –. MIT

Skater (🥉14 · ⭐ 1.1K · 💀) - Python Library for Model Interpretation/Explanations. ❗️UPL-1.0

Attribution Priors (🥉13 · ⭐ 120 · 💀) - Tools for training explainable models using.. MIT

bias-detector (🥉13 · ⭐ 43 · 💤) - Bias Detector is a python package for detecting bias in machine.. MIT

contextual-ai (🥉12 · ⭐ 86 · 💀) - Contextual AI adds explainability to different stages of.. Apache-2

Vector Similarity Search (ANN)

Libraries for Approximate Nearest Neighbor Search and Vector Indexing/Similarity Search.

🔗 ANN Benchmarks ( ⭐ 5.1K) - Benchmarks of approximate nearest neighbor libraries in Python.

Milvus (🥇42 · ⭐ 32K) - Milvus is a high-performance, cloud-native vector database built for.. Apache-2

GitHub (👨‍💻 300 · 🔀 3K · 📥 310K · 📋 13K - 5% open · ⏱️ 13.02.2025):
```
	git clone https://github.com/milvus-io/milvus
```
PyPi (📥 1.5M / month · 📦 210 · ⏱️ 22.01.2025):
```
	pip install pymilvus
```
Docker Hub (📥 68M · ⭐ 71 · ⏱️ 13.02.2025):
```
	docker pull milvusdb/milvus
```

Faiss (🥇41 · ⭐ 33K) - A library for efficient similarity search and clustering of dense vectors. MIT

GitHub (👨‍💻 210 · 🔀 3.7K · 📦 4.4K · 📋 2.6K - 9% open · ⏱️ 07.02.2025):

	git clone https://github.com/facebookresearch/faiss

PyPi (📥 1.5M / month · 📦 210 · ⏱️ 22.01.2025):

	pip install pymilvus

Conda (📥 2.2M · ⏱️ 20.12.2024):

	conda install -c conda-forge faiss

Annoy (🥈34 · ⭐ 13K · 💤) - Approximate Nearest Neighbors in C++/Python optimized for memory.. Apache-2

GitHub (👨‍💻 88 · 🔀 1.2K · 📦 4.7K · 📋 410 - 15% open · ⏱️ 29.07.2024):

	git clone https://github.com/spotify/annoy

PyPi (📥 800K / month · 📦 200 · ⏱️ 14.06.2023):

	pip install annoy

Conda (📥 640K · ⏱️ 05.09.2024):

	conda install -c conda-forge python-annoy

hnswlib (🥈31 · ⭐ 4.5K · 💤) - Header-only C++/python library for fast approximate nearest.. Apache-2

GitHub (👨‍💻 72 · 🔀 670 · 📦 7.6K · 📋 410 - 60% open · ⏱️ 17.06.2024):

	git clone https://github.com/nmslib/hnswlib

PyPi (📥 650K / month · 📦 130 · ⏱️ 03.12.2023):

	pip install hnswlib

Conda (📥 310K · ⏱️ 20.11.2024):

	conda install -c conda-forge hnswlib

USearch (🥈31 · ⭐ 2.5K) - Fast Open-Source Search & Clustering engine for Vectors & Strings in.. Apache-2

GitHub (👨‍💻 63 · 🔀 160 · 📥 51K · 📦 160 · 📋 200 - 40% open · ⏱️ 07.02.2025):

	git clone https://github.com/unum-cloud/usearch

PyPi (📥 190K / month · 📦 27 · ⏱️ 29.12.2024):

	pip install usearch

npm (📥 9.4K / month · 📦 15 · ⏱️ 23.01.2025):

	npm install usearch

Docker Hub (📥 170 · ⭐ 1 · ⏱️ 21.11.2024):

	docker pull unum/usearch

NMSLIB (🥉30 · ⭐ 3.4K) - Non-Metric Space Library (NMSLIB): An efficient similarity search.. Apache-2

GitHub (👨‍💻 49 · 🔀 450 · 📦 1.3K · 📋 440 - 20% open · ⏱️ 21.09.2024):

	git clone https://github.com/nmslib/nmslib

PyPi (📥 430K / month · 📦 63 · ⏱️ 03.02.2021):

	pip install nmslib

Conda (📥 180K · ⏱️ 09.09.2024):

	conda install -c conda-forge nmslib

PyNNDescent (🥉28 · ⭐ 910) - A Python nearest neighbor descent for approximate nearest neighbors. BSD-2

GitHub (👨‍💻 29 · 🔀 110 · 📦 9.4K · 📋 140 - 52% open · ⏱️ 10.11.2024):

	git clone https://github.com/lmcinnes/pynndescent

PyPi (📥 1.8M / month · 📦 160 · ⏱️ 17.06.2024):

	pip install pynndescent

Conda (📥 2.2M · ⏱️ 14.12.2024):

	conda install -c conda-forge pynndescent

NGT (🥉24 · ⭐ 1.3K) - Nearest Neighbor Search with Neighborhood Graph and Tree for High-.. Apache-2

GitHub (👨‍💻 16 · 🔀 120 · 📋 140 - 14% open · ⏱️ 29.01.2025):

	git clone https://github.com/yahoojapan/NGT

PyPi (📥 4.4K / month · 📦 12 · ⏱️ 23.01.2025):

	pip install ngt

Show 4 hidden projects...

NearPy (🥉21 · ⭐ 770 · 💀) - Python framework for fast (approximated) nearest neighbour search in.. MIT

Magnitude (🥉20 · ⭐ 1.6K · 💀) - A fast, efficient universal vector embedding utility package. MIT

N2 (🥉20 · ⭐ 570 · 💀) - TOROS N2 - lightweight approximate Nearest Neighbor library which runs.. Apache-2

PySparNN (🥉11 · ⭐ 920 · 💀) - Approximate Nearest Neighbor Search for Sparse Data in Python!. BSD-3

Probabilistics & Statistics

Libraries providing capabilities for probabilistic programming/reasoning, bayesian inference, gaussian processes, or statistics.

PyMC3 (🥇41 · ⭐ 8.9K) - Bayesian Modeling and Probabilistic Programming in Python. Apache-2

GitHub (👨‍💻 510 · 🔀 2K · 📥 2K · 📦 5.5K · 📋 3.4K - 10% open · ⏱️ 10.02.2025):
```
	git clone https://github.com/pymc-devs/pymc
```
PyPi (📥 300K / month · 📦 190 · ⏱️ 31.05.2024):
```
	pip install pymc3
```
Conda (📥 640K · ⏱️ 21.12.2024):
```
	conda install -c conda-forge pymc3
```

tensorflow-probability (🥇37 · ⭐ 4.3K) - Probabilistic reasoning and statistical analysis in.. Apache-2

GitHub (👨‍💻 500 · 🔀 1.1K · 📦 3 · 📋 1.5K - 48% open · ⏱️ 29.01.2025):

	git clone https://github.com/tensorflow/probability

PyPi (📥 1.3M / month · 📦 620 · ⏱️ 08.11.2024):

	pip install tensorflow-probability

Conda (📥 170K · ⏱️ 27.05.2024):

	conda install -c conda-forge tensorflow-probability

patsy (🥇34 · ⭐ 960) - Describing statistical models in Python using symbolic formulas. BSD-2

GitHub (👨‍💻 22 · 🔀 100 · 📦 120K · 📋 160 - 46% open · ⏱️ 11.02.2025):

	git clone https://github.com/pydata/patsy

PyPi (📥 15M / month · 📦 530 · ⏱️ 12.11.2024):

	pip install patsy

Conda (📥 14M · ⏱️ 10.12.2024):

	conda install -c conda-forge patsy

pgmpy (🥈33 · ⭐ 2.8K) - Python Library for learning (Structure and Parameter), inference.. MIT

GitHub (👨‍💻 140 · 🔀 720 · 📥 590 · 📦 1.4K · 📋 970 - 30% open · ⏱️ 12.02.2025):

	git clone https://github.com/pgmpy/pgmpy

PyPi (📥 83K / month · 📦 53 · ⏱️ 09.08.2024):

	pip install pgmpy

Pyro (🥈32 · ⭐ 8.7K) - Deep universal probabilistic programming with Python and PyTorch. Apache-2

GitHub (👨‍💻 160 · 🔀 990 · 📋 1.1K - 23% open · ⏱️ 05.02.2025):

	git clone https://github.com/pyro-ppl/pyro

PyPi (📥 350K / month · 📦 190 · ⏱️ 02.06.2024):

	pip install pyro-ppl

Conda (📥 220K · ⏱️ 18.12.2024):

	conda install -c conda-forge pyro-ppl

GPyTorch (🥈32 · ⭐ 3.6K) - A highly efficient implementation of Gaussian Processes in PyTorch. MIT

GitHub (👨‍💻 140 · 🔀 560 · 📦 2.6K · 📋 1.4K - 27% open · ⏱️ 07.02.2025):

	git clone https://github.com/cornellius-gp/gpytorch

PyPi (📥 290K / month · 📦 190 · ⏱️ 29.01.2025):

	pip install gpytorch

Conda (📥 190K · ⏱️ 02.02.2025):

	conda install -c conda-forge gpytorch

pandas-ta (🥈31 · ⭐ 5.8K · 💤) - Technical Analysis Indicators - Pandas TA is an easy to use.. MIT

GitHub (👨‍💻 45 · 🔀 1.1K · 📦 4.7K · 📋 610 - 17% open · ⏱️ 24.06.2024):

	git clone https://github.com/twopirllc/pandas-ta

PyPi (📥 160K / month · 📦 140 · ⏱️ 28.07.2021):

	pip install pandas-ta

Conda (📥 24K · ⏱️ 19.01.2025):

	conda install -c conda-forge pandas-ta

hmmlearn (🥈30 · ⭐ 3.1K) - Hidden Markov Models in Python, with scikit-learn like API. BSD-3

GitHub (👨‍💻 49 · 🔀 740 · 📦 3.1K · 📋 450 - 15% open · ⏱️ 31.10.2024):

	git clone https://github.com/hmmlearn/hmmlearn

PyPi (📥 140K / month · 📦 92 · ⏱️ 31.10.2024):

	pip install hmmlearn

Conda (📥 340K · ⏱️ 31.10.2024):

	conda install -c conda-forge hmmlearn

emcee (🥈30 · ⭐ 1.5K) - The Python ensemble sampling toolkit for affine-invariant MCMC. MIT

GitHub (👨‍💻 75 · 🔀 430 · 📦 2.8K · 📋 300 - 19% open · ⏱️ 01.02.2025):

	git clone https://github.com/dfm/emcee

PyPi (📥 120K / month · 📦 440 · ⏱️ 19.04.2024):

	pip install emcee

Conda (📥 390K · ⏱️ 13.12.2024):

	conda install -c conda-forge emcee

GPflow (🥉29 · ⭐ 1.9K) - Gaussian processes in TensorFlow. Apache-2

GitHub (👨‍💻 84 · 🔀 440 · 📦 730 · 📋 840 - 18% open · ⏱️ 29.01.2025):

	git clone https://github.com/GPflow/GPflow

PyPi (📥 65K / month · 📦 35 · ⏱️ 17.06.2024):

	pip install gpflow

Conda (📥 40K · ⏱️ 26.06.2024):

	conda install -c conda-forge gpflow

bambi (🥉28 · ⭐ 1.1K) - BAyesian Model-Building Interface (Bambi) in Python. MIT

GitHub (👨‍💻 46 · 🔀 130 · 📦 180 · 📋 430 - 20% open · ⏱️ 21.01.2025):

	git clone https://github.com/bambinos/bambi

PyPi (📥 32K / month · 📦 14 · ⏱️ 21.12.2024):

	pip install bambi

Conda (📥 44K · ⏱️ 23.12.2024):

	conda install -c conda-forge bambi

SALib (🥉28 · ⭐ 900) - Sensitivity Analysis Library in Python. Contains Sobol, Morris, FAST, and.. MIT

GitHub (👨‍💻 50 · 🔀 240 · 📦 1.4K · 📋 340 - 15% open · ⏱️ 06.01.2025):

	git clone https://github.com/SALib/SALib

PyPi (📥 220K / month · 📦 130 · ⏱️ 19.08.2024):

	pip install salib

Conda (📥 200K · ⏱️ 27.12.2024):

	conda install -c conda-forge salib

pomegranate (🥉27 · ⭐ 3.4K) - Fast, flexible and easy to use probabilistic modelling in Python. MIT

GitHub (👨‍💻 75 · 🔀 590 · 📋 790 - 3% open · ⏱️ 07.02.2025):

	git clone https://github.com/jmschrei/pomegranate

PyPi (📥 17K / month · 📦 67 · ⏱️ 07.02.2025):

	pip install pomegranate

Conda (📥 190K · ⏱️ 22.12.2024):

	conda install -c conda-forge pomegranate

PyStan (🥉27 · ⭐ 350 · 💤) - PyStan, a Python interface to Stan, a platform for statistical.. ISC

GitHub (👨‍💻 14 · 🔀 60 · 📦 10K · 📋 200 - 6% open · ⏱️ 03.07.2024):

	git clone https://github.com/stan-dev/pystan

PyPi (📥 780K / month · 📦 160 · ⏱️ 03.07.2024):

	pip install pystan

Conda (📥 3M · ⏱️ 16.06.2023):

	conda install -c conda-forge pystan

scikit-posthocs (🥉26 · ⭐ 360) - Multiple Pairwise Comparisons (Post Hoc) Tests in Python. MIT

GitHub (👨‍💻 15 · 🔀 40 · 📥 66 · 📦 990 · 📋 63 - 4% open · ⏱️ 18.12.2024):

	git clone https://github.com/maximtrp/scikit-posthocs

PyPi (📥 88K / month · 📦 63 · ⏱️ 18.12.2024):

	pip install scikit-posthocs

Conda (📥 1M · ⏱️ 23.12.2024):

	conda install -c conda-forge scikit-posthocs

Orbit (🥉24 · ⭐ 1.9K · 💤) - A Python package for Bayesian forecasting with object-oriented.. Apache-2

GitHub (👨‍💻 20 · 🔀 140 · 📦 67 · 📋 400 - 12% open · ⏱️ 10.07.2024):

	git clone https://github.com/uber/orbit

PyPi (📥 10K / month · 📦 1 · ⏱️ 01.04.2024):

	pip install orbit-ml

TorchUncertainty (🥉23 · ⭐ 350) - Open-source framework for uncertainty and deep.. Apache-2

GitHub (👨‍💻 10 · 🔀 24 · 📋 40 - 27% open · ⏱️ 21.01.2025):

	git clone https://github.com/ENSTA-U2IS-AI/torch-uncertainty

PyPi (📥 1.5K / month · 📦 3 · ⏱️ 21.01.2025):

	pip install torch-uncertainty

Baal (🥉21 · ⭐ 880 · 💤) - Bayesian active learning library for research and industrial usecases. Apache-2

GitHub (👨‍💻 23 · 🔀 87 · 📦 66 · 📋 110 - 17% open · ⏱️ 27.06.2024):

	git clone https://github.com/baal-org/baal

PyPi (📥 1.6K / month · 📦 2 · ⏱️ 11.06.2024):

	pip install baal

Conda (📥 12K · ⏱️ 12.06.2023):

	conda install -c conda-forge baal

pyhsmm (🥉21 · ⭐ 560) - Bayesian inference in HSMMs and HMMs. MIT

GitHub (👨‍💻 14 · 🔀 170 · 📦 34 · 📋 100 - 39% open · ⏱️ 25.01.2025):

	git clone https://github.com/mattjj/pyhsmm

PyPi (📥 260 / month · 📦 1 · ⏱️ 10.05.2017):

	pip install pyhsmm

Show 5 hidden projects...

filterpy (🥈31 · ⭐ 3.4K · 💀) - Python Kalman filtering and optimal estimation library. Implements.. MIT

pingouin (🥈30 · ⭐ 1.7K) - Statistical package in Python based on Pandas. ❗️GPL-3.0

Edward (🥉27 · ⭐ 4.8K · 💀) - A probabilistic programming language in TensorFlow. Deep.. Apache-2

Funsor (🥉20 · ⭐ 240 · 💀) - Functional tensors for probabilistic programming. Apache-2

ZhuSuan (🥉15 · ⭐ 2.2K · 💀) - A probabilistic programming library for Bayesian deep learning,.. MIT

Adversarial Robustness

Libraries for testing the robustness of machine learning models against attacks with adversarial/malicious examples.

ART (🥇34 · ⭐ 5K) - Adversarial Robustness Toolbox (ART) - Python Library for Machine Learning.. MIT

GitHub (👨‍💻 140 · 🔀 1.2K · 📦 660 · 📋 900 - 2% open · ⏱️ 04.02.2025):
```
	git clone https://github.com/Trusted-AI/adversarial-robustness-toolbox
```
PyPi (📥 26K / month · 📦 20 · ⏱️ 22.01.2025):
```
	pip install adversarial-robustness-toolbox
```

Conda (📥 64K · ⏱️ 22.01.2025):

	conda install -c conda-forge adversarial-robustness-toolbox

TextAttack (🥈27 · ⭐ 3.1K · 💤) - TextAttack is a Python framework for adversarial attacks, data.. MIT

GitHub (👨‍💻 66 · 🔀 400 · 📦 350 · 📋 290 - 23% open · ⏱️ 25.07.2024):

	git clone https://github.com/QData/TextAttack

PyPi (📥 5.4K / month · 📦 11 · ⏱️ 11.03.2024):

	pip install textattack

Conda (📥 9.5K · ⏱️ 16.06.2023):

	conda install -c conda-forge textattack

Foolbox (🥈27 · ⭐ 2.8K · 💤) - A Python toolbox to create adversarial examples that fool neural.. MIT

GitHub (👨‍💻 35 · 🔀 420 · 📦 670 · 📋 380 - 7% open · ⏱️ 04.03.2024):

	git clone https://github.com/bethgelab/foolbox

PyPi (📥 4.5K / month · 📦 14 · ⏱️ 04.03.2024):

	pip install foolbox

Conda (📥 17K · ⏱️ 16.06.2023):

	conda install -c conda-forge foolbox

Show 6 hidden projects...

CleverHans (🥈29 · ⭐ 6.2K · 💀) - An adversarial example library for constructing attacks,.. MIT

advertorch (🥉22 · ⭐ 1.3K · 💀) - A Toolbox for Adversarial Robustness Research. ❗️GPL-3.0

robustness (🥉20 · ⭐ 930 · 💀) - A library for experimenting with, training and evaluating neural.. MIT

AdvBox (🥉19 · ⭐ 1.4K · 💀) - Advbox is a toolbox to generate adversarial examples that fool.. Apache-2

textflint (🥉17 · ⭐ 640 · 💀) - Unified Multilingual Robustness Evaluation Toolkit for.. ❗️GPL-3.0

Adversary (🥉15 · ⭐ 400 · 💀) - Tool to generate adversarial text examples and test machine.. MIT

GPU & Accelerator Utilities

Libraries that require and make use of CUDA/GPU or other accelerator hardware capabilities to optimize machine learning tasks.

CuPy (🥇38 · ⭐ 9.9K) - NumPy & SciPy for GPU. MIT

GitHub (👨‍💻 400 · 🔀 860 · 📥 190K · 📦 2.5K · 📋 2.4K - 25% open · ⏱️ 12.02.2025):
```
	git clone https://github.com/cupy/cupy
```
PyPi (📥 36K / month · 📦 270 · ⏱️ 22.08.2024):
```
	pip install cupy
```
Conda (📥 5.8M · ⏱️ 18.10.2024):
```
	conda install -c conda-forge cupy
```
Docker Hub (📥 76K · ⭐ 13 · ⏱️ 22.08.2024):
```
	docker pull cupy/cupy
```

optimum (🥇36 · ⭐ 2.7K) - Accelerate inference and training of Transformers, Diffusers, TIMM.. Apache-2

GitHub (👨‍💻 140 · 🔀 490 · 📦 4.5K · 📋 880 - 46% open · ⏱️ 12.02.2025):

	git clone https://github.com/huggingface/optimum

PyPi (📥 920K / month · 📦 200 · ⏱️ 30.01.2025):

	pip install optimum

Conda (📥 33K · ⏱️ 29.05.2024):

	conda install -c conda-forge optimum

cuDF (🥈35 · ⭐ 8.7K) - cuDF - GPU DataFrame Library. Apache-2

GitHub (👨‍💻 300 · 🔀 920 · 📦 59 · 📋 6.8K - 15% open · ⏱️ 13.02.2025):

	git clone https://github.com/rapidsai/cudf

PyPi (📥 3.8K / month · 📦 22 · ⏱️ 01.06.2020):

	pip install cudf

Apex (🥈31 · ⭐ 8.5K) - A PyTorch Extension: Tools for easy mixed precision and distributed.. BSD-3

GitHub (👨‍💻 130 · 🔀 1.4K · 📦 3K · 📋 1.3K - 58% open · ⏱️ 05.02.2025):

	git clone https://github.com/NVIDIA/apex

Conda (📥 440K · ⏱️ 04.11.2024):

	conda install -c conda-forge nvidia-apex

cuML (🥈31 · ⭐ 4.4K) - cuML - RAPIDS Machine Learning Library. Apache-2

GitHub (👨‍💻 180 · 🔀 550 · 📋 2.6K - 35% open · ⏱️ 13.02.2025):

	git clone https://github.com/rapidsai/cuml

PyPi (📥 3.4K / month · 📦 14 · ⏱️ 01.06.2020):

	pip install cuml

PyCUDA (🥈30 · ⭐ 1.9K) - CUDA integration for Python, plus shiny features. MIT

GitHub (👨‍💻 82 · 🔀 290 · 📦 3.4K · 📋 280 - 30% open · ⏱️ 07.02.2025):

	git clone https://github.com/inducer/pycuda

PyPi (📥 87K / month · 📦 170 · ⏱️ 07.02.2025):

	pip install pycuda

Conda (📥 880K · ⏱️ 17.08.2024):

	conda install -c conda-forge pycuda

ArrayFire (🥈28 · ⭐ 4.6K) - ArrayFire: a general purpose GPU library. BSD-3

GitHub (👨‍💻 95 · 🔀 530 · 📥 7.8K · 📋 1.7K - 19% open · ⏱️ 13.02.2025):

	git clone https://github.com/arrayfire/arrayfire

PyPi (📥 3.8K / month · 📦 10 · ⏱️ 22.02.2022):

	pip install arrayfire

cuGraph (🥉27 · ⭐ 1.9K) - cuGraph - RAPIDS Graph Analytics Library. Apache-2

GitHub (👨‍💻 120 · 🔀 310 · 📋 1.8K - 10% open · ⏱️ 12.02.2025):

	git clone https://github.com/rapidsai/cugraph

PyPi (📥 310 / month · 📦 4 · ⏱️ 01.06.2020):

	pip install cugraph

Conda (📥 27K · ⏱️ 16.06.2023):

	conda install -c conda-forge libcugraph

DALI (🥉25 · ⭐ 5.3K) - A GPU-accelerated library containing highly optimized building blocks.. Apache-2

GitHub (👨‍💻 95 · 🔀 630 · 📋 1.7K - 15% open · ⏱️ 13.02.2025):

	git clone https://github.com/NVIDIA/DALI

Vulkan Kompute (🥉23 · ⭐ 2.1K) - General purpose GPU compute framework built on Vulkan to.. Apache-2

GitHub (👨‍💻 31 · 🔀 150 · 📥 620 · 📋 220 - 32% open · ⏱️ 11.02.2025):
```
	git clone https://github.com/KomputeProject/kompute
```
PyPi (📥 450 / month · ⏱️ 20.01.2024):
```
	pip install kp
```

Merlin (🥉21 · ⭐ 800 · 💤) - NVIDIA Merlin is an open source library providing end-to-end GPU-.. Apache-2

GitHub (👨‍💻 32 · 🔀 120 · 📋 460 - 46% open · ⏱️ 22.07.2024):

	git clone https://github.com/NVIDIA-Merlin/Merlin

PyPi (📥 14K / month · 📦 1 · ⏱️ 29.08.2023):

	pip install merlin-core

Show 9 hidden projects...

gpustat (🥈30 · ⭐ 4.1K · 💀) - A simple command-line utility for querying and monitoring GPU status. MIT

GPUtil (🥉25 · ⭐ 1.2K · 💀) - A Python module for getting the GPU status from NVIDA GPUs using.. MIT

scikit-cuda (🥉24 · ⭐ 990 · 💀) - Python interface to GPU-powered libraries. BSD-3

py3nvml (🥉22 · ⭐ 240 · 💀) - Python 3 Bindings for NVML library. Get NVIDIA GPU status inside.. BSD-3

BlazingSQL (🥉20 · ⭐ 1.9K · 💀) - BlazingSQL is a lightweight, GPU accelerated, SQL engine for.. Apache-2

nvidia-ml-py3 (🥉18 · ⭐ 140 · 💤) - Python 3 Bindings for the NVIDIA Management Library. BSD-3

cuSignal (🥉16 · ⭐ 720 · 💀) - GPU accelerated signal processing. Apache-2

SpeedTorch (🥉16 · ⭐ 680 · 💀) - Library for faster pinned CPU - GPU transfer in Pytorch. MIT

ipyexperiments (🥉16 · ⭐ 210 · 💀) - Automatic GPU+CPU memory profiling, re-use and memory.. Apache-2

Tensorflow Utilities

Libraries that extend TensorFlow with additional capabilities.

TensorFlow Datasets (🥇39 · ⭐ 4.4K) - TFDS is a collection of datasets ready to use with.. Apache-2

GitHub (👨‍💻 430 · 🔀 1.6K · 📦 22K · 📋 1.5K - 47% open · ⏱️ 10.02.2025):
```
	git clone https://github.com/tensorflow/datasets
```
PyPi (📥 1.9M / month · 📦 330 · ⏱️ 30.10.2024):
```
	pip install tensorflow-datasets
```

Conda (📥 43K · ⏱️ 16.06.2023):

	conda install -c conda-forge tensorflow-datasets

TFX (🥇33 · ⭐ 2.1K) - TFX is an end-to-end platform for deploying production ML pipelines. Apache-2

GitHub (👨‍💻 190 · 🔀 710 · 📦 1.7K · 📋 1.1K - 22% open · ⏱️ 12.12.2024):

	git clone https://github.com/tensorflow/tfx

PyPi (📥 51K / month · 📦 17 · ⏱️ 11.12.2024):

	pip install tfx

tensorflow-hub (🥈32 · ⭐ 3.5K) - A library for transfer learning by reusing parts of.. Apache-2

GitHub (👨‍💻 110 · 🔀 1.7K · 📋 710 - 1% open · ⏱️ 17.01.2025):

	git clone https://github.com/tensorflow/hub

PyPi (📥 1.9M / month · 📦 300 · ⏱️ 30.01.2024):

	pip install tensorflow-hub

Conda (📥 110K · ⏱️ 05.02.2025):

	conda install -c conda-forge tensorflow-hub

TF Addons (🥈31 · ⭐ 1.7K · 💤) - Useful extra functionality for TensorFlow 2.x maintained.. Apache-2

GitHub (👨‍💻 210 · 🔀 610 · 📋 990 - 9% open · ⏱️ 15.04.2024):

	git clone https://github.com/tensorflow/addons

PyPi (📥 2.3M / month · 📦 370 · ⏱️ 28.11.2023):

	pip install tensorflow-addons

TF Model Optimization (🥈29 · ⭐ 1.5K) - A toolkit to optimize ML models for deployment for.. Apache-2

GitHub (👨‍💻 87 · 🔀 320 · 📋 400 - 57% open · ⏱️ 10.02.2025):

	git clone https://github.com/tensorflow/model-optimization

PyPi (📥 450K / month · 📦 45 · ⏱️ 08.02.2024):

	pip install tensorflow-model-optimization

TensorFlow I/O (🥈29 · ⭐ 720 · 💤) - Dataset, streaming, and file system extensions.. Apache-2

GitHub (👨‍💻 110 · 🔀 290 · 📋 660 - 44% open · ⏱️ 01.07.2024):

	git clone https://github.com/tensorflow/io

PyPi (📥 1M / month · 📦 61 · ⏱️ 01.07.2024):

	pip install tensorflow-io

TensorFlow Transform (🥉27 · ⭐ 990) - Input pipeline framework. Apache-2

GitHub (👨‍💻 29 · 🔀 220 · 📋 220 - 17% open · ⏱️ 07.02.2025):

	git clone https://github.com/tensorflow/transform

PyPi (📥 390K / month · 📦 18 · ⏱️ 28.10.2024):

	pip install tensorflow-transform

Neural Structured Learning (🥉24 · ⭐ 980) - Training neural models with structured signals. Apache-2

GitHub (👨‍💻 39 · 🔀 190 · 📦 500 · 📋 69 - 1% open · ⏱️ 29.01.2025):

	git clone https://github.com/tensorflow/neural-structured-learning

PyPi (📥 6K / month · 📦 3 · ⏱️ 29.07.2022):

	pip install neural-structured-learning

Saliency (🥉22 · ⭐ 960 · 💤) - Framework-agnostic implementation for state-of-the-art.. Apache-2

GitHub (👨‍💻 18 · 🔀 190 · 📦 120 · 📋 39 - 30% open · ⏱️ 20.03.2024):

	git clone https://github.com/PAIR-code/saliency

PyPi (📥 17K / month · 📦 8 · ⏱️ 20.03.2024):

	pip install saliency

TF Compression (🥉21 · ⭐ 870) - Data compression in TensorFlow. Apache-2

GitHub (👨‍💻 24 · 🔀 250 · 📋 100 - 10% open · ⏱️ 29.01.2025):

	git clone https://github.com/tensorflow/compression

PyPi (📥 3.2K / month · 📦 2 · ⏱️ 02.02.2024):

	pip install tensorflow-compression

TensorFlow Cloud (🥉21 · ⭐ 380) - The TensorFlow Cloud repository provides APIs that.. Apache-2

GitHub (👨‍💻 28 · 🔀 90 · 📋 100 - 73% open · ⏱️ 29.01.2025):

	git clone https://github.com/tensorflow/cloud

PyPi (📥 33K / month · 📦 7 · ⏱️ 17.06.2021):

	pip install tensorflow-cloud

Show 5 hidden projects...

tensor2tensor (🥇33 · ⭐ 16K · 💀) - Library of deep learning models and datasets designed.. Apache-2

Keras-Preprocessing (🥈29 · ⭐ 1K · 💀) - Utilities for working with image data, text data, and.. MIT

efficientnet (🥉26 · ⭐ 2.1K · 💀) - Implementation of EfficientNet model. Keras and.. Apache-2

TensorNets (🥉21 · ⭐ 1K · 💀) - High level network definitions with pre-trained weights in.. MIT

tffm (🥉18 · ⭐ 780 · 💀) - TensorFlow implementation of an arbitrary order Factorization Machine. MIT

Jax Utilities

Libraries that extend Jax with additional capabilities.

equinox (🥇33 · ⭐ 2.2K) - Elegant easy-to-use neural networks + scientific computing in.. Apache-2

GitHub (👨‍💻 59 · 🔀 150 · 📦 1K · 📋 500 - 33% open · ⏱️ 04.02.2025):
```
	git clone https://github.com/patrick-kidger/equinox
```
PyPi (📥 1.3M / month · 📦 200 · ⏱️ 24.12.2024):
```
	pip install equinox
```

evojax (🥉19 · ⭐ 870 · 💤) - EvoJAX: Hardware-accelerated Neuroevolution. Apache-2

GitHub (👨‍💻 14 · 🔀 95 · 📦 27 · 📋 37 - 54% open · ⏱️ 27.06.2024):

	git clone https://github.com/google/evojax

PyPi (📥 1.6K / month · 📦 6 · ⏱️ 18.06.2024):

	pip install evojax

Conda (📥 35K · ⏱️ 18.06.2024):

	conda install -c conda-forge evojax

Show 1 hidden projects...

jaxdf (🥉12 · ⭐ 120) - A JAX-based research framework for writing differentiable.. ❗️LGPL-3.0

Sklearn Utilities

Libraries that extend scikit-learn with additional capabilities.

MLxtend (🥇35 · ⭐ 5K) - A library of extension and helper modules for Pythons data analysis.. BSD-3

GitHub (👨‍💻 110 · 🔀 880 · 📦 18K · 📋 500 - 29% open · ⏱️ 26.01.2025):
```
	git clone https://github.com/rasbt/mlxtend
```
PyPi (📥 690K / month · 📦 200 · ⏱️ 26.01.2025):
```
	pip install mlxtend
```
Conda (📥 340K · ⏱️ 26.01.2025):
```
	conda install -c conda-forge mlxtend
```

scikit-learn-intelex (🥇35 · ⭐ 1.3K) - Extension for Scikit-learn is a seamless way to speed.. Apache-2

GitHub (👨‍💻 84 · 🔀 180 · 📦 13K · 📋 250 - 20% open · ⏱️ 11.02.2025):

	git clone https://github.com/intel/scikit-learn-intelex

PyPi (📥 67K / month · 📦 55 · ⏱️ 16.01.2025):

	pip install scikit-learn-intelex

Conda (📥 450K · ⏱️ 13.02.2025):

	conda install -c conda-forge scikit-learn-intelex

imbalanced-learn (🥈34 · ⭐ 6.9K) - A Python Package to Tackle the Curse of Imbalanced.. MIT

GitHub (👨‍💻 86 · 🔀 1.3K · 📋 620 - 7% open · ⏱️ 23.12.2024):

	git clone https://github.com/scikit-learn-contrib/imbalanced-learn

PyPi (📥 14M / month · 📦 480 · ⏱️ 20.12.2024):

	pip install imbalanced-learn

Conda (📥 660K · ⏱️ 20.12.2024):

	conda install -c conda-forge imbalanced-learn

category_encoders (🥈32 · ⭐ 2.4K) - A library of sklearn compatible categorical variable.. BSD-3

GitHub (👨‍💻 70 · 🔀 400 · 📦 2.7K · 📋 300 - 14% open · ⏱️ 19.01.2025):

	git clone https://github.com/scikit-learn-contrib/category_encoders

PyPi (📥 1.5M / month · 📦 300 · ⏱️ 19.01.2025):

	pip install category_encoders

Conda (📥 300K · ⏱️ 19.01.2025):

	conda install -c conda-forge category_encoders

scikit-lego (🥈28 · ⭐ 1.3K) - Extra blocks for scikit-learn pipelines. MIT

GitHub (👨‍💻 68 · 🔀 120 · 📦 170 · 📋 330 - 12% open · ⏱️ 19.01.2025):

	git clone https://github.com/koaning/scikit-lego

PyPi (📥 25K / month · 📦 13 · ⏱️ 17.12.2024):

	pip install scikit-lego

Conda (📥 63K · ⏱️ 10.07.2024):

	conda install -c conda-forge scikit-lego

scikit-opt (🥉25 · ⭐ 5.4K · 💤) - Genetic Algorithm, Particle Swarm Optimization, Simulated.. MIT

GitHub (👨‍💻 24 · 🔀 990 · 📦 250 · 📋 180 - 37% open · ⏱️ 23.06.2024):

	git clone https://github.com/guofei9987/scikit-opt

PyPi (📥 3.9K / month · 📦 15 · ⏱️ 14.01.2022):

	pip install scikit-opt

iterative-stratification (🥉22 · ⭐ 860) - scikit-learn cross validators for iterative.. BSD-3

GitHub (👨‍💻 7 · 🔀 75 · 📦 550 · 📋 27 - 7% open · ⏱️ 12.10.2024):

	git clone https://github.com/trent-b/iterative-stratification

PyPi (📥 22K / month · 📦 15 · ⏱️ 12.10.2024):

	pip install iterative-stratification

dabl (🥉20 · ⭐ 730) - Data Analysis Baseline Library. BSD-3

GitHub (👨‍💻 24 · 🔀 100 · ⏱️ 07.08.2024):

	git clone https://github.com/amueller/dabl

PyPi (📥 5.5K / month · 📦 3 · ⏱️ 16.12.2024):

	pip install dabl

scikit-tda (🥉18 · ⭐ 530 · 💤) - Topological Data Analysis for Python. MIT

GitHub (👨‍💻 6 · 🔀 55 · 📦 70 · 📋 22 - 18% open · ⏱️ 19.07.2024):

	git clone https://github.com/scikit-tda/scikit-tda

PyPi (📥 1.4K / month · ⏱️ 19.07.2024):

	pip install scikit-tda

DESlib (🥉18 · ⭐ 480 · 💤) - A Python library for dynamic classifier and ensemble selection. BSD-3

GitHub (👨‍💻 17 · 🔀 100 · 📋 160 - 11% open · ⏱️ 15.04.2024):

	git clone https://github.com/scikit-learn-contrib/DESlib

PyPi (📥 2.8K / month · 📦 3 · ⏱️ 12.04.2024):

	pip install deslib

Show 9 hidden projects...

scikit-survival (🥈32 · ⭐ 1.2K) - Survival analysis built on top of scikit-learn. ❗️GPL-3.0

fancyimpute (🥈27 · ⭐ 1.3K · 💀) - Multivariate imputation and matrix completion.. Apache-2

scikit-multilearn (🥈27 · ⭐ 930 · 💀) - A scikit-learn based module for multi-label et. al… BSD-2

sklearn-crfsuite (🥉26 · ⭐ 430 · 💀) - scikit-learn inspired API for CRFsuite. MIT

sklearn-contrib-lightning (🥉23 · ⭐ 1.7K · 💀) - Large-scale linear classification, regression and.. BSD-3

skope-rules (🥉22 · ⭐ 630 · 💀) - machine learning with logical rules in Python. ❗️BSD-1-Clause

combo (🥉21 · ⭐ 650 · 💀) - (AAAI 20) A Python Toolbox for Machine Learning Model.. BSD-2 xgboost

celer (🥉20 · ⭐ 210) - Fast solver for L1-type problems: Lasso, sparse Logisitic regression,.. BSD-3

skggm (🥉18 · ⭐ 250 · 💀) - Scikit-learn compatible estimation of general graphical models. MIT

Pytorch Utilities

Libraries that extend Pytorch with additional capabilities.

accelerate (🥇41 · ⭐ 8.3K) - A simple way to launch, train, and use PyTorch models on.. Apache-2

GitHub (👨‍💻 320 · 🔀 1K · 📦 71K · 📋 1.7K - 7% open · ⏱️ 11.02.2025):
```
	git clone https://github.com/huggingface/accelerate
```
PyPi (📥 9.2M / month · 📦 1.8K · ⏱️ 17.01.2025):
```
	pip install accelerate
```

Conda (📥 300K · ⏱️ 20.01.2025):

	conda install -c conda-forge accelerate

PML (🥇34 · ⭐ 6.1K) - The easiest way to use deep metric learning in your application. Modular,.. MIT

GitHub (👨‍💻 46 · 🔀 660 · 📦 2.3K · 📋 520 - 14% open · ⏱️ 11.12.2024):

	git clone https://github.com/KevinMusgrave/pytorch-metric-learning

PyPi (📥 660K / month · 📦 55 · ⏱️ 11.12.2024):

	pip install pytorch-metric-learning

Conda (📥 12K · ⏱️ 16.06.2023):

	conda install -c metric-learning pytorch-metric-learning

tinygrad (🥇33 · ⭐ 28K) - You like pytorch? You like micrograd? You love tinygrad!. MIT

GitHub (👨‍💻 370 · 🔀 3.1K · 📦 160 · 📋 860 - 16% open · ⏱️ 13.02.2025):

	git clone https://github.com/geohot/tinygrad

torchdiffeq (🥇30 · ⭐ 5.8K) - Differentiable ODE solvers with full GPU support and.. MIT

GitHub (👨‍💻 21 · 🔀 930 · 📦 4.5K · 📋 220 - 33% open · ⏱️ 21.11.2024):
```
	git clone https://github.com/rtqichen/torchdiffeq
```
PyPi (📥 880K / month · 📦 120 · ⏱️ 21.11.2024):
```
	pip install torchdiffeq
```

Conda (📥 20K · ⏱️ 16.06.2023):

	conda install -c conda-forge torchdiffeq

torchsde (🥈29 · ⭐ 1.6K) - Differentiable SDE solvers with GPU support and efficient.. Apache-2

GitHub (👨‍💻 9 · 🔀 200 · 📦 4.5K · 📋 82 - 35% open · ⏱️ 30.12.2024):

	git clone https://github.com/google-research/torchsde

PyPi (📥 2.3M / month · 📦 37 · ⏱️ 26.09.2023):

	pip install torchsde

Conda (📥 35K · ⏱️ 21.11.2024):

	conda install -c conda-forge torchsde

Pytorch Toolbelt (🥈26 · ⭐ 1.5K · 📈) - PyTorch extensions for fast R&D prototyping and.. MIT

GitHub (👨‍💻 8 · 🔀 120 · 📥 140 · 📋 33 - 12% open · ⏱️ 06.02.2025):

	git clone https://github.com/BloodAxe/pytorch-toolbelt

PyPi (📥 11K / month · 📦 12 · ⏱️ 21.11.2024):

	pip install pytorch_toolbelt

torch-scatter (🥈25 · ⭐ 1.6K) - PyTorch Extension Library of Optimized Scatter Operations. MIT

GitHub (👨‍💻 32 · 🔀 180 · 📋 400 - 6% open · ⏱️ 10.01.2025):

	git clone https://github.com/rusty1s/pytorch_scatter

PyPi (📥 40K / month · 📦 150 · ⏱️ 06.10.2023):

	pip install torch-scatter

Conda (📥 740K · ⏱️ 13.02.2025):

	conda install -c conda-forge pytorch_scatter

EfficientNets (🥈25 · ⭐ 1.6K · 💤) - Pretrained EfficientNet, EfficientNet-Lite, MixNet,.. Apache-2

GitHub (👨‍💻 5 · 🔀 210 · 📦 280 · 📋 55 - 7% open · ⏱️ 13.06.2024):

	git clone https://github.com/rwightman/gen-efficientnet-pytorch

PyPi (📥 150K / month · 📦 4 · ⏱️ 08.07.2021):

	pip install geffnet

PyTorch Sparse (🥉24 · ⭐ 1K) - PyTorch Extension Library of Optimized Autograd Sparse Matrix.. MIT

GitHub (👨‍💻 46 · 🔀 150 · 📋 290 - 10% open · ⏱️ 10.01.2025):

	git clone https://github.com/rusty1s/pytorch_sparse

PyPi (📥 29K / month · 📦 120 · ⏱️ 06.10.2023):

	pip install torch-sparse

Conda (📥 700K · ⏱️ 05.11.2024):

	conda install -c conda-forge pytorch_sparse

pytorchviz (🥉19 · ⭐ 3.3K) - A small package to create visualizations of PyTorch execution graphs. MIT

GitHub (👨‍💻 6 · 🔀 280 · 📦 2.5K · 📋 72 - 47% open · ⏱️ 30.12.2024):

	git clone https://github.com/szagoruyko/pytorchviz

madgrad (🥉17 · ⭐ 800) - MADGRAD Optimization Method. MIT

GitHub (👨‍💻 3 · 🔀 57 · 📦 100 · ⏱️ 27.01.2025):
```
	git clone https://github.com/facebookresearch/madgrad
```
PyPi (📥 3.3K / month · 📦 1 · ⏱️ 08.03.2022):
```
	pip install madgrad
```

Show 21 hidden projects...

pretrainedmodels (🥈29 · ⭐ 9.1K · 💀) - Pretrained ConvNets for pytorch: NASNet, ResNeXt,.. BSD-3

pytorch-summary (🥈28 · ⭐ 4K · 💀) - Model summary in PyTorch similar to model.summary() in.. MIT

lightning-flash (🥈28 · ⭐ 1.7K · 💀) - Your PyTorch AI Factory - Flash enables you to easily.. Apache-2

EfficientNet-PyTorch (🥈27 · ⭐ 8K · 💀) - A PyTorch implementation of EfficientNet. Apache-2

pytorch-optimizer (🥈27 · ⭐ 3.1K · 💀) - torch-optimizer – collection of optimizers for.. Apache-2

TabNet (🥈25 · ⭐ 2.7K · 💀) - PyTorch implementation of TabNet paper :.. MIT

Torchmeta (🥉24 · ⭐ 2K · 💀) - A collection of extensions and data-loaders for few-shot.. MIT

Higher (🥉24 · ⭐ 1.6K · 💀) - higher is a pytorch library allowing users to obtain higher.. Apache-2

micrograd (🥉22 · ⭐ 11K · 💀) - A tiny scalar-valued autograd engine and a neural net library.. MIT

SRU (🥉22 · ⭐ 2.1K · 💀) - Training RNNs as Fast as CNNs (https://arxiv.org/abs/1709.02755). MIT

Antialiased CNNs (🥉22 · ⭐ 1.7K · 💀) - pip install antialiased-cnns to improve stability and.. ❗️CC BY-NC-SA 4.0

AdaBound (🥉21 · ⭐ 2.9K · 💀) - An optimizer that trains as fast as Adam and as good as SGD. Apache-2

reformer-pytorch (🥉21 · ⭐ 2.2K · 💀) - Reformer, the efficient Transformer, in Pytorch. MIT

Poutyne (🥉21 · ⭐ 570) - A simplified framework and utilities for PyTorch. ❗️LGPL-3.0

Performer Pytorch (🥉20 · ⭐ 1.1K · 💀) - An implementation of Performer, a linear attention-.. MIT

Lambda Networks (🥉19 · ⭐ 1.5K · 💀) - Implementation of LambdaNetworks, a new approach to.. MIT

Torch-Struct (🥉18 · ⭐ 1.1K · 💀) - Fast, general, and tested differentiable structured.. MIT

Tensor Sensor (🥉18 · ⭐ 800 · 💀) - The goal of this library is to generate more helpful.. MIT

Tez (🥉17 · ⭐ 1.2K · 💀) - Tez is a super-simple and lightweight Trainer for PyTorch. It.. Apache-2

Pywick (🥉17 · ⭐ 400 · 💀) - High-level batteries-included neural network training library for.. MIT

TorchDrift (🥉15 · ⭐ 320 · 💀) - Drift Detection for your PyTorch Models. Apache-2

Database Clients

Libraries for connecting to, operating, and querying databases.

🔗 best-of-python - DB Clients ( ⭐ 3.8K) - Collection of database clients for python.

Others

scipy (🥇50 · ⭐ 13K) - Ecosystem of open-source software for mathematics, science, and engineering. BSD-3

GitHub (👨‍💻 1.7K · 🔀 5.2K · 📥 460K · 📦 1.2M · 📋 11K - 15% open · ⏱️ 13.02.2025):
```
	git clone https://github.com/scipy/scipy
```
PyPi (📥 130M / month · 📦 49K · ⏱️ 10.01.2025):
```
	pip install scipy
```
Conda (📥 58M · ⏱️ 11.01.2025):
```
	conda install -c conda-forge scipy
```

SymPy (🥇49 · ⭐ 13K) - A computer algebra system written in pure Python. BSD-3

GitHub (👨‍💻 1.4K · 🔀 4.5K · 📥 550K · 📦 220K · 📋 14K - 36% open · ⏱️ 11.02.2025):

	git clone https://github.com/sympy/sympy

PyPi (📥 35M / month · 📦 3.5K · ⏱️ 18.09.2024):

	pip install sympy

Conda (📥 7.8M · ⏱️ 07.01.2025):

	conda install -c conda-forge sympy

Streamlit (🥇46 · ⭐ 37K) - Streamlit A faster way to build and share data apps. Apache-2

GitHub (👨‍💻 310 · 🔀 3.2K · 📦 710K · 📋 4.9K - 20% open · ⏱️ 13.02.2025):

	git clone https://github.com/streamlit/streamlit

PyPi (📥 7.2M / month · 📦 3.2K · ⏱️ 04.02.2025):

	pip install streamlit

Gradio (🥇44 · ⭐ 36K) - Wrap UIs around any model, share with anyone. Apache-2

GitHub (👨‍💻 500 · 🔀 2.7K · 📦 56K · 📋 5.3K - 8% open · ⏱️ 12.02.2025):

	git clone https://github.com/gradio-app/gradio

PyPi (📥 6.9M / month · 📦 1K · ⏱️ 12.02.2025):

	pip install gradio

carla (🥇38 · ⭐ 12K) - Open-source simulator for autonomous driving research. MIT

GitHub (👨‍💻 180 · 🔀 3.7K · 📦 930 · 📋 5.8K - 19% open · ⏱️ 13.02.2025):

	git clone https://github.com/carla-simulator/carla

PyPi (📥 11K / month · 📦 11 · ⏱️ 14.11.2023):

	pip install carla

PennyLane (🥇37 · ⭐ 2.5K) - PennyLane is a cross-platform Python library for quantum.. Apache-2

GitHub (👨‍💻 190 · 🔀 620 · 📥 100 · 📦 1.3K · 📋 1.5K - 22% open · ⏱️ 13.02.2025):

	git clone https://github.com/PennyLaneAI/PennyLane

PyPi (📥 75K / month · 📦 120 · ⏱️ 14.01.2025):

	pip install pennylane

Conda (📥 240K · ⏱️ 09.07.2024):

	conda install -c conda-forge pennylane

PyOD (🥇36 · ⭐ 8.8K) - A Python Library for Outlier and Anomaly Detection, Integrating Classical.. BSD-2

GitHub (👨‍💻 62 · 🔀 1.4K · 📦 4.7K · 📋 370 - 60% open · ⏱️ 22.12.2024):

	git clone https://github.com/yzhao062/pyod

PyPi (📥 610K / month · 📦 120 · ⏱️ 22.12.2024):

	pip install pyod

Conda (📥 140K · ⏱️ 06.09.2024):

	conda install -c conda-forge pyod

Autograd (🥇36 · ⭐ 7.1K) - Efficiently computes derivatives of NumPy code. MIT

GitHub (👨‍💻 60 · 🔀 910 · 📦 11K · 📋 430 - 41% open · ⏱️ 11.02.2025):

	git clone https://github.com/HIPS/autograd

PyPi (📥 3.6M / month · 📦 280 · ⏱️ 22.08.2024):

	pip install autograd

Conda (📥 510K · ⏱️ 13.12.2024):

	conda install -c conda-forge autograd

Datasette (🥈35 · ⭐ 9.8K) - An open source multi-tool for exploring and publishing data. Apache-2

GitHub (👨‍💻 81 · 🔀 700 · 📥 70 · 📦 1.4K · 📋 1.9K - 32% open · ⏱️ 13.02.2025):

	git clone https://github.com/simonw/datasette

PyPi (📥 220K / month · 📦 440 · ⏱️ 06.02.2025):

	pip install datasette

Conda (📥 54K · ⏱️ 30.11.2024):

	conda install -c conda-forge datasette

agate (🥈35 · ⭐ 1.2K) - A Python data analysis library that is optimized for humans instead of.. MIT

GitHub (👨‍💻 53 · 🔀 150 · 📦 4.5K · 📋 650 - 0% open · ⏱️ 29.01.2025):

	git clone https://github.com/wireservice/agate

PyPi (📥 15M / month · 📦 54 · ⏱️ 29.01.2025):

	pip install agate

Conda (📥 290K · ⏱️ 29.01.2025):

	conda install -c conda-forge agate

DeepChem (🥈34 · ⭐ 5.7K · 📉) - Democratizing Deep-Learning for Drug Discovery, Quantum.. MIT

GitHub (👨‍💻 260 · 🔀 1.7K · 📦 510 · 📋 1.9K - 35% open · ⏱️ 10.02.2025):

	git clone https://github.com/deepchem/deepchem

PyPi (📥 46K / month · 📦 16 · ⏱️ 10.02.2025):

	pip install deepchem

Conda (📥 110K · ⏱️ 05.04.2024):

	conda install -c conda-forge deepchem

hdbscan (🥈34 · ⭐ 2.9K) - A high performance implementation of HDBSCAN clustering. BSD-3

GitHub (👨‍💻 96 · 🔀 500 · 📦 4.8K · 📋 530 - 67% open · ⏱️ 08.02.2025):

	git clone https://github.com/scikit-learn-contrib/hdbscan

PyPi (📥 630K / month · 📦 350 · ⏱️ 18.11.2024):

	pip install hdbscan

Conda (📥 2.4M · ⏱️ 12.10.2024):

	conda install -c conda-forge hdbscan

Pythran (🥈33 · ⭐ 2K) - Ahead of Time compiler for numeric kernels. BSD-3

GitHub (👨‍💻 73 · 🔀 190 · 📦 3.1K · 📋 880 - 15% open · ⏱️ 09.02.2025):

	git clone https://github.com/serge-sans-paille/pythran

PyPi (📥 330K / month · 📦 21 · ⏱️ 31.10.2024):

	pip install pythran

Conda (📥 1M · ⏱️ 23.11.2024):

	conda install -c conda-forge pythran

River (🥈32 · ⭐ 5.2K) - Online machine learning in Python. BSD-3

GitHub (👨‍💻 120 · 🔀 550 · 📦 640 · 📋 620 - 19% open · ⏱️ 03.02.2025):

	git clone https://github.com/online-ml/river

PyPi (📥 84K / month · 📦 64 · ⏱️ 25.11.2024):

	pip install river

Conda (📥 100K · ⏱️ 06.10.2023):

	conda install -c conda-forge river

tensorly (🥈32 · ⭐ 1.6K · 📉) - TensorLy: Tensor Learning in Python. BSD-2

GitHub (👨‍💻 70 · 🔀 290 · 📦 890 · 📋 270 - 21% open · ⏱️ 31.01.2025):

	git clone https://github.com/tensorly/tensorly

PyPi (📥 57K / month · 📦 99 · ⏱️ 12.11.2024):

	pip install tensorly

Conda (📥 370K · ⏱️ 10.06.2024):

	conda install -c conda-forge tensorly

pyjanitor (🥈32 · ⭐ 1.4K) - Clean APIs for data cleaning. Python implementation of R package.. MIT

GitHub (👨‍💻 110 · 🔀 170 · 📦 840 · 📋 570 - 19% open · ⏱️ 06.02.2025):

	git clone https://github.com/pyjanitor-devs/pyjanitor

PyPi (📥 94K / month · 📦 36 · ⏱️ 04.12.2024):

	pip install pyjanitor

Conda (📥 240K · ⏱️ 05.12.2024):

	conda install -c conda-forge pyjanitor

PaddleHub (🥈31 · ⭐ 13K) - Awesome pre-trained models toolkit based on PaddlePaddle... Apache-2

GitHub (👨‍💻 70 · 🔀 2.1K · 📥 840 · 📦 1.8K · 📋 1.3K - 44% open · ⏱️ 07.08.2024):

	git clone https://github.com/PaddlePaddle/PaddleHub

PyPi (📥 5K / month · 📦 7 · ⏱️ 20.09.2023):

	pip install paddlehub

datalad (🥈31 · ⭐ 560) - Keep code, data, containers under control with git and git-annex. MIT

GitHub (👨‍💻 57 · 🔀 110 · 📦 450 · 📋 4K - 13% open · ⏱️ 15.12.2024):

	git clone https://github.com/datalad/datalad

PyPi (📥 17K / month · 📦 99 · ⏱️ 15.12.2024):

	pip install datalad

Conda (📥 800K · ⏱️ 20.11.2024):

	conda install -c conda-forge datalad

pyopencl (🥈30 · ⭐ 1.1K · 📉) - OpenCL integration for Python, plus shiny features. MIT

GitHub (👨‍💻 97 · 🔀 240 · 📦 2.1K · 📋 360 - 21% open · ⏱️ 05.02.2025):

	git clone https://github.com/inducer/pyopencl

PyPi (📥 91K / month · 📦 180 · ⏱️ 22.01.2025):

	pip install pyopencl

Conda (📥 1.6M · ⏱️ 22.01.2025):

	conda install -c conda-forge pyopencl

causalml (🥈29 · ⭐ 5.2K) - Uplift modeling and causal inference with machine learning.. Apache-2

GitHub (👨‍💻 64 · 🔀 780 · 📦 250 · 📋 400 - 13% open · ⏱️ 07.02.2025):

	git clone https://github.com/uber/causalml

PyPi (📥 49K / month · 📦 2 · ⏱️ 01.10.2024):

	pip install causalml

anomalib (🥈29 · ⭐ 4K) - An anomaly detection library comprising state-of-the-art algorithms.. Apache-2

GitHub (👨‍💻 83 · 🔀 700 · 📥 18K · 📦 130 · 📋 970 - 15% open · ⏱️ 11.02.2025):

	git clone https://github.com/openvinotoolkit/anomalib

PyPi (📥 27K / month · 📦 5 · ⏱️ 09.01.2025):

	pip install anomalib

dstack (🥈29 · ⭐ 1.7K · 📉) - dstack is a lightweight, open-source alternative to Kubernetes &.. MPL-2.0

GitHub (👨‍💻 49 · 🔀 170 · 📦 18 · 📋 1.2K - 8% open · ⏱️ 13.02.2025):

	git clone https://github.com/dstackai/dstack

PyPi (📥 7.6K / month · ⏱️ 13.02.2025):

	pip install dstack

Prince (🥈29 · ⭐ 1.3K) - Multivariate exploratory data analysis in Python PCA, CA, MCA, MFA,.. MIT

GitHub (👨‍💻 16 · 🔀 180 · 📦 660 · ⏱️ 05.01.2025):

	git clone https://github.com/MaxHalford/prince

PyPi (📥 160K / month · 📦 19 · ⏱️ 04.01.2025):

	pip install prince

Conda (📥 23K · ⏱️ 16.06.2023):

	conda install -c conda-forge prince-factor-analysis

Trax (🥈28 · ⭐ 8.2K) - Trax Deep Learning with Clear Code and Speed. Apache-2

GitHub (👨‍💻 80 · 🔀 820 · 📦 210 · 📋 250 - 49% open · ⏱️ 07.02.2025):

	git clone https://github.com/google/trax

PyPi (📥 4.2K / month · 📦 1 · ⏱️ 26.10.2021):

	pip install trax

TabPy (🥈28 · ⭐ 1.6K) - Execute Python code on the fly and display results in Tableau visualizations:. MIT

GitHub (👨‍💻 51 · 🔀 600 · 📦 190 · 📋 320 - 6% open · ⏱️ 25.11.2024):

	git clone https://github.com/tableau/TabPy

PyPi (📥 6.5K / month · 📦 2 · ⏱️ 25.11.2024):

	pip install tabpy

Conda (📥 4.9K · ⏱️ 16.06.2023):

	conda install -c anaconda tabpy-client

adapter-transformers (🥉27 · ⭐ 2.6K) - A Unified Library for Parameter-Efficient and Modular.. Apache-2 huggingface

GitHub (👨‍💻 15 · 🔀 360 · 📦 160 · 📋 400 - 11% open · ⏱️ 28.01.2025):

	git clone https://github.com/Adapter-Hub/adapter-transformers

PyPi (📥 4.6K / month · 📦 12 · ⏱️ 07.07.2024):

	pip install adapter-transformers

avalanche (🥉27 · ⭐ 1.8K) - Avalanche: an End-to-End Library for Continual Learning based on.. MIT

GitHub (👨‍💻 81 · 🔀 300 · 📥 48 · 📦 120 · 📋 820 - 12% open · ⏱️ 29.10.2024):

	git clone https://github.com/ContinualAI/avalanche

PyPi (📥 1.5K / month · 📦 3 · ⏱️ 29.10.2024):

	pip install avalanche-lib

pycm (🥉27 · ⭐ 1.5K) - Multi-class confusion matrix library in Python. MIT

GitHub (👨‍💻 18 · 🔀 130 · 📦 370 · 📋 210 - 7% open · ⏱️ 13.01.2025):

	git clone https://github.com/sepandhaghighi/pycm

PyPi (📥 41K / month · 📦 24 · ⏱️ 14.01.2025):

	pip install pycm

pyclustering (🥉27 · ⭐ 1.2K · 💤) - pyclustering is a Python, C++ data mining library. BSD-3

GitHub (👨‍💻 26 · 🔀 250 · 📥 660 · 📦 800 · 📋 670 - 11% open · ⏱️ 08.02.2024):

	git clone https://github.com/annoviko/pyclustering

PyPi (📥 36K / month · 📦 32 · ⏱️ 25.11.2020):

	pip install pyclustering

Conda (📥 130K · ⏱️ 08.11.2024):

	conda install -c conda-forge pyclustering

metric-learn (🥉26 · ⭐ 1.4K) - Metric learning algorithms in Python. MIT

GitHub (👨‍💻 23 · 🔀 230 · 📦 450 · 📋 170 - 30% open · ⏱️ 03.08.2024):

	git clone https://github.com/scikit-learn-contrib/metric-learn

PyPi (📥 7K / month · 📦 7 · ⏱️ 09.10.2023):

	pip install metric-learn

Conda (📥 15K · ⏱️ 06.01.2025):

	conda install -c conda-forge metric-learn

Feature Engine (🥉25 · ⭐ 2K) - Feature engineering package with sklearn like functionality. BSD-3

GitHub (👨‍💻 49 · 🔀 320 · ⏱️ 31.08.2024):

	git clone https://github.com/solegalli/feature_engine

PyPi (📥 220K / month · 📦 180 · ⏱️ 22.01.2025):

	pip install feature_engine

Conda (📥 66K · ⏱️ 27.01.2025):

	conda install -c conda-forge feature_engine

AugLy (🥉24 · ⭐ 5K) - A data augmentations library for audio, image, text, and video. MIT

GitHub (👨‍💻 38 · 🔀 300 · 📦 160 · 📋 78 - 30% open · ⏱️ 05.02.2025):

	git clone https://github.com/facebookresearch/AugLy

PyPi (📥 3.3K / month · 📦 4 · ⏱️ 05.12.2023):

	pip install augly

BioPandas (🥉23 · ⭐ 720) - Working with molecular structures in pandas DataFrames. BSD-3

GitHub (👨‍💻 18 · 🔀 120 · 📦 340 · 📋 59 - 35% open · ⏱️ 01.08.2024):

	git clone https://github.com/rasbt/biopandas

PyPi (📥 11K / month · 📦 38 · ⏱️ 01.08.2024):

	pip install biopandas

Conda (📥 170K · ⏱️ 28.12.2024):

	conda install -c conda-forge biopandas

MONAILabel (🥉23 · ⭐ 660) - MONAI Label is an intelligent open source image labeling and.. Apache-2

GitHub (👨‍💻 65 · 🔀 200 · 📥 110K · 📋 540 - 24% open · ⏱️ 15.01.2025):

	git clone https://github.com/Project-MONAI/MONAILabel

PyPi (📥 2.4K / month · ⏱️ 01.10.2023):

	pip install monailabel-weekly

benchmark_VAE (🥉21 · ⭐ 1.9K · 💤) - Unifying Variational Autoencoder (VAE).. Apache-2

GitHub (👨‍💻 18 · 🔀 170 · 📦 37 · 📋 71 - 36% open · ⏱️ 17.07.2024):

	git clone https://github.com/clementchadebec/benchmark_VAE

PyPi (📥 1.5K / month · ⏱️ 06.09.2023):

	pip install pythae

SUOD (🥉21 · ⭐ 380 · 💤) - (MLSys 21) An Acceleration System for Large-scare Unsupervised.. BSD-2

GitHub (👨‍💻 3 · 🔀 49 · 📦 540 · 📋 15 - 80% open · ⏱️ 08.02.2024):

	git clone https://github.com/yzhao062/SUOD

PyPi (📥 12K / month · 📦 8 · ⏱️ 08.02.2024):

	pip install suod

pymdp (🥉20 · ⭐ 500) - A Python implementation of active inference for Markov Decision Processes. MIT

GitHub (👨‍💻 19 · 🔀 99 · 📦 17 · 📋 49 - 44% open · ⏱️ 06.02.2025):

	git clone https://github.com/infer-actively/pymdp

PyPi (📥 2.6K / month · ⏱️ 08.12.2022):

	pip install inferactively-pymdp

pykale (🥉20 · ⭐ 450) - Knowledge-Aware machine LEarning (KALE): accessible machine learning.. MIT

GitHub (👨‍💻 25 · 🔀 64 · 📦 6 · 📋 120 - 8% open · ⏱️ 24.09.2024):

	git clone https://github.com/pykale/pykale

PyPi (📥 370 / month · ⏱️ 12.04.2022):

	pip install pykale

NeuralCompression (🥉15 · ⭐ 530) - A collection of tools for neural compression enthusiasts. MIT

GitHub (👨‍💻 10 · 🔀 43 · 📋 71 - 8% open · ⏱️ 20.09.2024):

	git clone https://github.com/facebookresearch/NeuralCompression

PyPi (📥 250 / month · ⏱️ 03.10.2023):

	pip install neuralcompression

Show 27 hidden projects...

Cython BLIS (🥈32 · ⭐ 220) - Fast matrix-multiplication as a self-contained Python library no.. BSD-3

cleanlab (🥈31 · ⭐ 10K) - The standard data-centric AI package for data quality and machine.. ❗️AGPL-3.0

pysc2 (🥈28 · ⭐ 8.1K · 💀) - StarCraft II Learning Environment. Apache-2

alibi-detect (🥈28 · ⭐ 2.3K) - Algorithms for outlier, adversarial and drift detection. ❗️Intel

modAL (🥈28 · ⭐ 2.3K · 💀) - A modular active learning framework for Python. MIT

kmodes (🥈28 · ⭐ 1.3K · 💀) - Python implementations of the k-modes and k-prototypes clustering.. MIT

gplearn (🥉27 · ⭐ 1.7K · 💀) - Genetic Programming in Python, with a scikit-learn inspired API. BSD-3

PySwarms (🥉27 · ⭐ 1.3K · 💀) - A research toolkit for particle swarm optimization in Python. MIT

metricflow (🥉26 · ⭐ 1.2K) - MetricFlow allows you to define, build, and maintain metrics.. ❗Unlicensed

minisom (🥉25 · ⭐ 1.5K) - MiniSom is a minimalistic implementation of the Self Organizing.. ❗️CC-BY-3.0

findspark (🥉25 · ⭐ 520 · 💀) - Find pyspark to make it importable. BSD-3

pandas-ai (🥉24 · ⭐ 14K) - Chat with your database or your datalake (SQL, CSV, parquet)… ❗Unlicensed

Mars (🥉24 · ⭐ 2.7K · 💀) - Mars is a tensor-based unified framework for large-scale data.. Apache-2

AstroML (🥉23 · ⭐ 1.1K · 💀) - Machine learning, statistics, and data mining for astronomy.. BSD-2

opyrator (🥉22 · ⭐ 3.1K · 💀) - Turns your machine learning code into microservices with web API,.. MIT

mlens (🥉22 · ⭐ 850 · 💀) - ML-Ensemble high performance ensemble learning. MIT

vecstack (🥉22 · ⭐ 690 · 💀) - Python package for stacking (machine learning technique). MIT

impyute (🥉21 · ⭐ 360 · 💀) - Data imputations library to preprocess datasets with missing data. MIT

StreamAlert (🥉20 · ⭐ 2.9K · 💀) - StreamAlert is a serverless, realtime data analysis.. Apache-2

rrcf (🥉20 · ⭐ 500 · 💀) - Implementation of the Robust Random Cut Forest algorithm for anomaly.. MIT

apricot (🥉20 · ⭐ 500 · 💀) - apricot implements submodular optimization for the purpose of.. MIT

scikit-rebate (🥉20 · ⭐ 420 · 💀) - A scikit-learn-compatible Python implementation of.. MIT

baikal (🥉19 · ⭐ 590 · 💀) - A graph-based functional API for building complex scikit-learn.. BSD-3

KD-Lib (🥉16 · ⭐ 620 · 💀) - A Pytorch Knowledge Distillation library for benchmarking and.. MIT

pandas-ml (🥉16 · ⭐ 320 · 💀) - pandas, scikit-learn, xgboost and seaborn integration. BSD-3

nylon (🥉14 · ⭐ 83 · 💀) - An intelligent, flexible grammar of machine learning. MIT

traingenerator (🥉13 · ⭐ 1.4K · 💀) - A web app to generate template code for machine learning. MIT

Related Resources

Papers With Code: Discover ML papers, code, and evaluation tables.
Sotabench: Discover & compare open-source ML models.
Google Dataset Search: Dataset search engine by Google.
Dataset List: List of the biggest ML datasets from across the web.
Awesome Public Datasets: A topic-centric list of open datasets.
Best-of lists: Discover other best-of lists with awesome open-source projects on all kinds of topics.
best-of-python-dev: A ranked list of awesome python developer tools and libraries.
best-of-web-python: A ranked list of awesome python libraries for web development.

Contribution

Contributions are encouraged and always welcome! If you like to add or update projects, choose one of the following ways:

Open an issue by selecting one of the provided categories from the issue page and fill in the requested information.
Modify the projects.yaml with your additions or changes, and submit a pull request. This can also be done directly via the Github UI.

If you like to contribute to or share suggestions regarding the project metadata collection or markdown generation, please refer to the best-of-generator repository. If you like to create your own best-of list, we recommend to follow this guide.

For more information on how to add or update projects, please read the contribution guidelines. By participating in this project, you agree to abide by its Code of Conduct.

Best-of Machine Learning with Python

Contents

Explanation

Machine Learning Frameworks

Data Visualization

Text Data & NLP

Image Data

Graph Data

Audio Data

Geospatial Data

Financial Data

Time Series Data

Medical Data

Tabular Data

Optical Character Recognition

Data Containers & Structures

Data Loading & Extraction

Web Scraping & Crawling

Data Pipelines & Streaming

Distributed Machine Learning

Hyperparameter Optimization & AutoML

Reinforcement Learning

Recommender Systems

Privacy Machine Learning

Workflow & Experiment Tracking

Model Serialization & Deployment

Model Interpretability

Vector Similarity Search (ANN)

Probabilistics & Statistics

Adversarial Robustness

GPU & Accelerator Utilities

Tensorflow Utilities

Jax Utilities

Sklearn Utilities

Pytorch Utilities

Database Clients

Others

Related Resources

Contribution

License

Описание

Конвейеры