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 5.1M 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.
Contents
- Machine Learning Frameworks 64 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 6 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 55 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 (🥇56 · ⭐ 200K) - An Open Source Machine Learning Framework for Everyone. Apache-2
-
GitHub (👨💻 5K · 🔀 75K · 📦 540K · 📋 42K - 4% open · ⏱️ 30.10.2025):
git clone https://github.com/tensorflow/tensorflow- PyPi (📥 26M / month · 📦 9.6K · ⏱️ 13.08.2025):pip install tensorflow- Conda (📥 6M · ⏱️ 27.10.2025):conda install -c conda-forge tensorflow- Docker Hub (📥 81M · ⭐ 2.8K · ⏱️ 30.10.2025):docker pull tensorflow/tensorflow
scikit-learn (🥇53 · ⭐ 64K) - scikit-learn: machine learning in Python. BSD-3
XGBoost (🥇46 · ⭐ 28K) - Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or.. Apache-2
PaddlePaddle (🥇46 · ⭐ 23K) - PArallel Distributed Deep LEarning: Machine Learning.. Apache-2
pytorch-lightning (🥇45 · ⭐ 30K) - Pretrain, finetune ANY AI model of ANY size on 1 or.. Apache-2
StatsModels (🥇45 · ⭐ 11K) - Statsmodels: statistical modeling and econometrics in Python. BSD-3
LightGBM (🥈42 · ⭐ 18K) - A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT,.. MIT
Catboost (🥈42 · ⭐ 8.6K) - A fast, scalable, high performance Gradient Boosting on Decision.. Apache-2
einops (🥈35 · ⭐ 9.2K) - Flexible and powerful tensor operations for readable and reliable code.. MIT
Jina (🥈33 · ⭐ 22K · 💤) - Build multimodal AI applications with cloud-native stack. Apache-2
-
GitHub (👨💻 180 · 🔀 2.2K · ⏱️ 24.03.2025):
git clone https://github.com/jina-ai/jina- PyPi (📥 120K / month · 📦 29 · ⏱️ 24.03.2025):pip install jina- Conda (📥 110K · ⏱️ 22.04.2025):conda install -c conda-forge jina-core- Docker Hub (📥 1.8M · ⭐ 9 · ⏱️ 24.03.2025):docker pull jinaai/jina
Thinc (🥈33 · ⭐ 2.9K · 💤) - A refreshing functional take on deep learning, compatible with your.. MIT
tensorflow-upstream (🥉31 · ⭐ 700) - TensorFlow ROCm port. Apache-2
Determined (🥉26 · ⭐ 3.2K · 💤) - Determined is an open-source machine learning.. Apache-2
Neural Network Libraries (🥉26 · ⭐ 2.8K) - Neural Network Libraries. Apache-2
Towhee (🥉23 · ⭐ 3.4K · 💤) - Towhee is a framework that is dedicated to making neural data.. Apache-2
Runhouse (🥉21 · ⭐ 1.1K) - Distribute and run AI workloads magically in Python, like PyTorch.. Apache-2
chefboost (🥉19 · ⭐ 480) - A Lightweight Decision Tree Framework supporting regular algorithms:.. MIT
ThunderGBM (🥉18 · ⭐ 710 · 💤) - ThunderGBM: Fast GBDTs and Random Forests on GPUs. Apache-2
Show 26 hidden projects...
- dlib (🥈40 · ⭐ 14K) - A toolkit for making real world machine learning and data analysis..
❗️BSL-1.0 - MXNet (🥈38 · ⭐ 21K · 💀) - Lightweight, Portable, Flexible Distributed/Mobile Deep..
Apache-2 - Theano (🥈37 · ⭐ 10K · 💀) - Theano was a Python library that allows you to define, optimize, and..
BSD-3 - MindsDB (🥈33 · ⭐ 37K) - Federated query engine for AI - The only MCP Server youll ever need.
❗️ICU - Vowpal Wabbit (🥈33 · ⭐ 8.6K · 💀) - Vowpal Wabbit is a machine learning system which pushes the..
BSD-3 - Chainer (🥈33 · ⭐ 5.9K · 💀) - A flexible framework of neural networks for deep learning.
MIT - Turi Create (🥉32 · ⭐ 11K · 💀) - Turi Create simplifies the development of custom machine..
BSD-3 - tensorpack (🥉32 · ⭐ 6.3K · 💀) - A Neural Net Training Interface on TensorFlow, with..
Apache-2 - 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 - SHOGUN (🥉26 · ⭐ 3.1K · 💀) - Unified and efficient Machine Learning.
BSD-3 - ktrain (🥉26 · ⭐ 1.3K · 💀) - ktrain is a Python library that makes deep learning and AI..
Apache-2 - NeuPy (🥉25 · ⭐ 740 · 💀) - NeuPy is a Tensorflow based python library for prototyping and building..
MIT - xLearn (🥉24 · ⭐ 3.1K · 💀) - High performance, easy-to-use, and scalable machine learning (ML)..
Apache-2 - EvaDB (🥉24 · ⭐ 2.7K · 💀) - Database system for AI-powered apps.
Apache-2 - neon (🥉22 · ⭐ 3.9K · 💀) - Intel Nervana reference deep learning framework committed to best..
Apache-2 - ThunderSVM (🥉22 · ⭐ 1.6K · 💀) - ThunderSVM: A Fast SVM Library on GPUs and CPUs.
Apache-2 - Torchbearer (🥉22 · ⭐ 640 · 💀) - torchbearer: A model fitting library for PyTorch.
MIT - mace (🥉21 · ⭐ 5K · 💀) - MACE is a deep learning inference framework optimized for mobile..
Apache-2 - Neural Tangents (🥉21 · ⭐ 2.4K · 💀) - Fast and Easy Infinite Neural Networks in Python.
Apache-2 - Objax (🥉20 · ⭐ 770 · 💀) - Objax is a machine learning framework that provides an Object..
Apache-2 - elegy (🥉19 · ⭐ 480 · 💀) - A High Level API for Deep Learning in JAX.
MIT - StarSpace (🥉16 · ⭐ 4K · 💀) - Learning embeddings for classification, retrieval and ranking.
MIT - nanodl (🥉14 · ⭐ 300 · 💀) - A Jax-based library for building transformers, includes..
MIT
README 内容较长，此处已截断，完整内容请查看 GitHub 仓库。