Awesome Python Data Science

Efficiently computes derivatives of NumPy code.

Flax is a neural network library for JAX that is designed for flexibility.

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

Fit interpretable models. Explain blackbox machine learning.

A Python scikit for building and analyzing recommender systems

Image processing in Python

A scikit-learn compatible neural network library that wraps PyTorch

A system for quickly generating training data with weak supervision

Uplift modeling and causal inference with machine learning algorithms

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 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 platform for Reasoning systems (Reinforcement Learning, Contextual Bandits, etc.)

Plotting library for IPython/Jupyter notebooks

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

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

With Holoviews, your data visualizes itself.

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

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

POT : Python Optimal Transport

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).

A Python package for manipulating 2-dimensional tabular data structures

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

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

Clean PyTorch implementations of imitation and reward learning algorithms

Lightweight and extensible compatibility layer between dataframe libraries!

A distributed task scheduler for Dask

Python audio and music signal processing library

Hyper-parameter optimization for sklearn

Hyperparameter Experiments with TensorFlow and Keras