Awesome Python Data Science

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

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

cuML - RAPIDS Machine Learning Library

Python framework for creating, editing, and running Noisy Intermediate-Scale Quantum (NISQ) circuits.

Lightning ⚡️ fast forecasting with statistical and econometric models.

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

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

A highly efficient implementation of Gaussian Processes in PyTorch

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

C++ library for audio and music analysis, description and synthesis, including Python bindings

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

Plotting library for IPython/Jupyter notebooks

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

Accelerated deep learning R&D

Python Toolkit for Causal and Probabilistic Reasoning

The machine learning toolkit for time series analysis in Python

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

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/

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

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

POT : Python Optimal Transport

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

Adaptive Experimentation Platform

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.

Graph Neural Networks with Keras and Tensorflow 2.

Optax is a gradient processing and optimization library for JAX.

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

🏕️ Reproducible development environment for humans and agents

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

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

A Python package for manipulating 2-dimensional tabular data structures

Python library for interactive topic model visualization. Port of the R LDAvis package.

Clean PyTorch implementations of imitation and reward learning algorithms

Python audio and music signal processing library

Lightweight and extensible compatibility layer between dataframe libraries!

A distributed task scheduler for Dask

Hyper-parameter optimization for sklearn

Hyperparameter Experiments with TensorFlow and Keras

The Python ensemble sampling toolkit for affine-invariant MCMC

Simple text to phones converter for multiple languages

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

PySAL: Python Spatial Analysis Library Meta-Package

Supply a wrapper ``StockDataFrame`` based on the ``pandas.DataFrame`` with inline stock statistics/indicators support.

Clean APIs for data cleaning. Python implementation of R package Janitor

Python interface for igraph