Awesome Python Data Science

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

Fit interpretable models. Explain blackbox machine learning.

Image processing in Python

A scikit-learn compatible neural network library that wraps PyTorch

Uplift modeling and causal inference with machine learning algorithms

A system for quickly generating training data with weak supervision

Python tools for geographic data

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 light-weight, flexible, and expressive statistical data testing library

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.

Missing data visualization module for Python.

N-D labeled arrays and datasets in Python

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.

A highly efficient implementation of Gaussian Processes in PyTorch

Vizro is a low-code toolkit for building high-quality data visualization apps.

a library for audio and music analysis

A platform for Reasoning systems (Reinforcement Learning, Contextual Bandits, etc.)

OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.

Bayesian optimization in PyTorch

Models, data loaders and abstractions for language processing, powered by PyTorch

A standard API for multi-agent reinforcement learning environments, with popular reference environments and related utilities

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.

Accelerated deep learning R&D

Python Toolkit for Causal and Probabilistic Reasoning

Tensorforce: a TensorFlow library for applied reinforcement learning

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 datastore for time series and tick data

NSGA2, NSGA3, R-NSGA3, MOEAD, Genetic Algorithms (GA), Differential Evolution (DE), CMAES, PSO

With Holoviews, your data visualizes itself.

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.

POT : Python Optimal Transport

Pandas integration with sklearn

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

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

Adaptive Experimentation Platform

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.

Algorithms for outlier, adversarial and drift detection

Fast numerical array expression evaluator for Python, NumPy, Pandas, PyTables and more

Automatic architecture search and hyperparameter optimization for PyTorch

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