Awesome Python Data Science

The Python ensemble sampling toolkit for affine-invariant MCMC

Keras community contributions

Distributed Deep learning with Keras & Spark

open-source feature selection repository in python

Spearmint Bayesian optimization codebase

Simple text to phones converter for multiple languages

MLBox is a powerful Automated Machine Learning python library.

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

PySAL: Python Spatial Analysis Library Meta-Package

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

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

OpenMMLab Foundational Library for Training Deep Learning Models

moDel Agnostic Language for Exploration and eXplanation

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

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

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

Python implementation of CMA-ES

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

Model analysis tools for TensorFlow

Scalable machine 🤖 learning for time series forecasting.

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.

[CONTRIBUTORS WELCOME] Generalized Additive Models in Python

Pretrained model hub for Keras 3.

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

Scalable Machine Learning with Dask

Factorization machines in python

Distributed machine learning platform

The Classical Language Toolkit

OpenFE: automated feature generation with expert-level performance

python partial dependence plot toolbox

Universal model exchange and serialization format for decision tree forests

Code for "High-Precision Model-Agnostic Explanations" paper

A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)

Library for exploring and validating machine learning data

ML powered analytics engine for outlier detection and root cause analysis.

TensorFlow implementation of an arbitrary order Factorization Machine