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.5M 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 62 projects
- Data Visualization 54 projects
- Text Data & NLP 102 projects
- Image Data 64 projects
- Graph Data 36 projects
- Audio Data 29 projects
- Geospatial Data 22 projects
- Financial Data 25 projects
- Time Series Data 30 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 1 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 23 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 · ⭐ 190K) - An Open Source Machine Learning Framework for Everyone. Apache-2
-
GitHub (👨💻 4.6K · 🔀 74K · 📦 380K · 📋 40K - 7% open · ⏱️ 06.06.2024):
git clone https://github.com/tensorflow/tensorflow
-
PyPi (📥 21M / month · 📦 7.4K · ⏱️ 09.03.2024):
pip install tensorflow
-
Conda (📥 4.7M · ⏱️ 24.05.2024):
conda install -c conda-forge tensorflow
-
Docker Hub (📥 77M · ⭐ 2.4K · ⏱️ 06.06.2024):
docker pull tensorflow/tensorflow
scikit-learn (🥇52 · ⭐ 59K) - scikit-learn: machine learning in Python. BSD-3
PaddlePaddle (🥇45 · ⭐ 22K) - PArallel Distributed Deep LEarning: Machine Learning.. Apache-2
StatsModels (🥇45 · ⭐ 9.7K) - Statsmodels: statistical modeling and econometrics in Python. BSD-3
pytorch-lightning (🥈44 · ⭐ 27K) - Pretrain, finetune and deploy AI models on multiple.. Apache-2
XGBoost (🥈44 · ⭐ 26K) - Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or.. Apache-2
LightGBM (🥈43 · ⭐ 16K) - A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT,.. MIT
Catboost (🥈41 · ⭐ 7.8K) - A fast, scalable, high performance Gradient Boosting on Decision.. Apache-2
Jina (🥈39 · ⭐ 20K) - Build multimodal AI applications with cloud-native stack. Apache-2
-
GitHub (👨💻 180 · 🔀 2.2K · 📦 1.7K · 📋 2K - 1% open · ⏱️ 16.05.2024):
git clone https://github.com/jina-ai/jina
-
PyPi (📥 40K / month · 📦 27 · ⏱️ 10.04.2024):
pip install jina
-
Conda (📥 65K · ⏱️ 16.06.2023):
conda install -c conda-forge jina-core
-
Docker Hub (📥 1.3M · ⭐ 8 · ⏱️ 16.05.2024):
docker pull jinaai/jina