Awesome Python Data Science

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

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

matplotlib: plotting with Python

🦉 Data Versioning and ML Experiments

Statistical data visualization in Python

1 Line of code data quality profiling & exploratory data analysis for Pandas and Spark DataFrames.

Low-code framework for building custom LLMs, neural networks, and other AI models

Statsmodels: statistical modeling and econometrics in Python

Cleanlab's open-source library is the standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.

Always know what to expect from your data.

Fast and Accurate ML in 3 Lines of Code

Modin: Scale your Pandas workflows by changing a single line of code

A unified framework for machine learning with time series

A Python Automated Machine Learning tool that optimizes machine learning pipelines using genetic programming.

Open-source, low-code AutoML platform for Python. PyCaret 4.0: sklearn-native engine + React control plane.

cuDF - GPU DataFrame Library

Deep learning library featuring a higher-level API for TensorFlow.

A python library for user-friendly forecasting and anomaly detection on time series.

Automatic extraction of relevant features from time series:

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.

Out-of-Core hybrid Apache Arrow/NumPy DataFrame for Python, ML, visualization and exploration of big tabular data at a billion rows per second 🚀

Evidently is ​​an open-source ML and LLM observability framework. Evaluate, test, and monitor any AI-powered system or data pipeline. From tabular data to Gen AI. 100+ metrics.

An open source python library for automated feature engineering

A Python Package to Tackle the Curse of Imbalanced Datasets in Machine Learning

A system for quickly generating training data with weak supervision

A library of extension and helper modules for Python's data analysis and machine learning libraries.

Lightning ⚡️ fast forecasting with statistical and econometric models.

Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML validation needs, enabling to thoroughly test your data and models from research to production.

A Python package for interactive geospatial analysis and visualization with Google Earth Engine.

Python library that makes it easy for data scientists to create charts.

Visualize and compare datasets, target values and associations, with one line of code.

High performance, easy-to-use, and scalable machine learning (ML) package, including linear model (LR), factorization machines (FM), and field-aware factorization machines (FFM) for Python and CLI interface.

Shōgun

StellarGraph - Machine Learning on Graphs

Deepnote is a drop-in replacement for Jupyter with an AI-first design, sleek UI, new blocks, and native data integrations. Use Python, R, and SQL locally in your favorite IDE, then scale to Deepnote cloud for real-time collaboration, Deepnote agent, and deployable data apps. https://deepnote.com/

A library for debugging/inspecting machine learning classifiers and explaining their predictions

Apache Hamilton helps data scientists and engineers define testable, modular, self-documenting dataflows, that encode lineage/tracing and metadata. Runs and scales everywhere python does.

2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.

An intuitive library to add plotting functionality to scikit-learn objects.

Feature engineering and selection open-source Python library compatible with sklearn.

Open-source low code data preparation library in python. Collect, clean and visualization your data in python with a few lines of code.

Keras community contributions

Python library for time series forecasting using scikit-learn compatible models, statistical methods, and foundation models

MLBox is a powerful Automated Machine Learning python library.

moDel Agnostic Language for Exploration and eXplanation

Transpile trained scikit-learn estimators to C, Java, JavaScript and others.

[CONTRIBUTORS WELCOME] Generalized Additive Models in Python

Easy pipelines for pandas DataFrames.

Pandas, Polars, Spark, and Snowpark DataFrame comparison for humans and more!

Engine for AI/ML/Data tracking, visualization, explainability, drift detection, and dashboards for Polyaxon.