Awesome Python Data Science

Hyper-parameter optimization for sklearn

Hyperparameter Experiments with TensorFlow and Keras

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

Mesh TensorFlow: Model Parallelism Made Easier

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

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

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

OpenFE: automated feature generation with expert-level performance

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

Library for exploring and validating machine learning data

:small_red_triangle: Ternary plotting library for python with matplotlib

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