Awesome Python Data Science

The Python ensemble sampling toolkit for affine-invariant MCMC

open-source feature selection repository in python

Keras community contributions

Spearmint Bayesian optimization codebase

Distributed Deep learning with Keras & Spark

Simple text to phones converter for multiple languages

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

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

MLBox is a powerful Automated Machine Learning python library.

PySAL: Python Spatial Analysis Library Meta-Package

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.

OpenMMLab Foundational Library for Training Deep Learning Models

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

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

Python implementation of CMA-ES

sqldf for pandas

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

Scalable machine 🤖 learning for time series forecasting.

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.

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

fastFM: A Library for Factorization Machines

Industry-strength Computer Vision workflows with Keras

[CONTRIBUTORS WELCOME] Generalized Additive Models in Python

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

Pretrained model hub for Keras 3.

An engine for high performance multi-agent environments with very large numbers of agents, along with a set of reference environments

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

Scalable Machine Learning with Dask

PettingZoo and Gymnasium bindings for popular reinforcement learning environments outside of Farama

The Classical Language Toolkit

Factorization machines in python

Distributed machine learning platform

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