Awesome Python Data Science

Natural Gradient Boosting for Probabilistic Prediction

A Python package for manipulating 2-dimensional tabular data structures

A flexible, intuitive and fast forecasting library

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

Deep learning with dynamic computation graphs in TensorFlow

Interpretability and explainability of data and machine learning models

Clean PyTorch implementations of imitation and reward learning algorithms

A high performance Python graph library implemented in Rust.

Painlessly create beautiful matplotlib plots.

Python audio and music signal processing library

A distributed task scheduler for Dask

An offline deep reinforcement learning library

Synthetic data generators for tabular and time-series data

Machine learning evaluation metrics, implemented in Python, R, Haskell, and MATLAB / Octave

Hyper-parameter optimization for sklearn

Hyperparameter Experiments with TensorFlow and Keras

Mesh TensorFlow: Model Parallelism Made Easier

Python implementations of the Boruta all-relevant feature selection method.

ThunderSVM: A Fast SVM Library on GPUs and CPUs

The Python ensemble sampling toolkit for affine-invariant MCMC

Keras community contributions

open-source feature selection repository in python

Distributed Deep learning with Keras & Spark

Spearmint Bayesian optimization codebase

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

MLBox is a powerful Automated Machine Learning python library.

C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.

OpenMMLab Foundational Library for Training Deep Learning Models

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

moDel Agnostic Language for Exploration and eXplanation

A Graph Neural Network Library in Jax

Python interface for igraph

No-code in the front, Python in the back. An open-source framework for creating data apps.

Metric learning algorithms in Python

A research toolkit for particle swarm optimization in Python

sqldf for pandas

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

Model analysis tools for TensorFlow

SMAC3: A Versatile Bayesian Optimization Package for Hyperparameter Optimization

Anomaly Detection and Correlation library

:chart_with_upwards_trend: Adaptive: parallel active learning of mathematical functions

Fast NumPy array functions written in C

PySpark + Scikit-learn = Sparkit-learn

An autoML framework & toolkit for machine learning on graphs.

fastFM: A Library for Factorization Machines

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments

Industry-strength Computer Vision workflows with Keras

allRank is a framework for training learning-to-rank neural models based on PyTorch.

Pretrained model hub for Keras 3.

A scikit-learn based module for multi-label et. al. classification