Awesome Python Data Science

mlpack: a fast, header-only C++ machine learning library

Deep Reinforcement Learning for Keras.

cuML - RAPIDS Machine Learning Library

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

Image augmentation library in Python for machine learning.

A Python implementation of LightFM, a hybrid recommendation algorithm.

Lightning ⚡️ fast forecasting with statistical and econometric models.

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

A collection of infrastructure and tools for research in neural network interpretability.

Visual analysis and diagnostic tools to facilitate machine learning model selection.

Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.

Scalable and user friendly neural :brain: forecasting algorithms.

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.

Fast, flexible and easy to use probabilistic modelling in Python.

Accelerated deep learning R&D

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

The machine learning toolkit for time series analysis in Python

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/

Data manipulation and transformation for audio signal processing, powered by PyTorch

A comprehensive set of fairness metrics for datasets and machine learning models, explanations for these metrics, and algorithms to mitigate bias in datasets and models.

Sequential model-based optimization with a `scipy.optimize` interface

POT : Python Optimal Transport

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

Learning to Rank in TensorFlow

Microsoft Distributed Machine Learning Toolkit

Algorithms for explaining machine learning models

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.

A modular active learning framework for Python

Source-to-Source Debuggable Derivatives in Pure Python

Optax is a gradient processing and optimization library for JAX.

Karate Club: An API Oriented Open-source Python Framework for Unsupervised Learning on Graphs (CIKM 2020)

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

Source code of PyGAD, a Python 3 library for building the genetic algorithm and training machine learning algorithms (Keras & PyTorch).

Automatically Visualize any dataset, any size with a single line of code. Created by Ram Seshadri. Collaborators Welcome. Permission Granted upon Request.

Natural Gradient Boosting for Probabilistic Prediction

Genetic Programming in Python, with a scikit-learn inspired API

Interpretability and explainability of data and machine learning models

Python audio and music signal processing library

Synthetic data generators for tabular and time-series data

Keras community contributions

MLBox is a powerful Automated Machine Learning python library.

TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.

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

OpenMMLab Foundational Library for Training Deep Learning Models