Awesome Python Data Science

A game theoretic approach to explain the output of any machine learning model.

LAVIS - A One-stop Library for Language-Vision Intelligence

Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.

A Python Automated Machine Learning tool that optimizes machine learning pipelines using genetic programming.

Automatic extraction of relevant features from time series:

Evidently is ​​an open-source ML and LLM observability framework. Evaluate, test, and monitor any AI-powered system or data pipeline. From tabular data to Gen AI. 100+ metrics.

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

A scikit-learn compatible neural network library that wraps PyTorch

A collection of infrastructure and tools for research in neural network interpretability.

Performance analysis of predictive (alpha) stock factors

Bayesian optimization in PyTorch

A library for debugging/inspecting machine learning classifiers and explaining their predictions

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.

Algorithms for outlier, adversarial and drift detection

2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.

Natural Gradient Boosting for Probabilistic Prediction

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

Synthetic data generators for tabular and time-series data

Python implementation of CMA-ES

python partial dependence plot toolbox

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

TensorFlow implementation of an arbitrary order Factorization Machine

Use evolutionary algorithms instead of gridsearch in scikit-learn

Easy pipelines for pandas DataFrames.

Machine Learning toolbox for Humans

🍇 GRAPE is a Rust/Python Graph Representation Learning library for Predictions and Evaluations

A library of metrics for evaluating recommender systems

Cleora AI is a general-purpose open-source model for efficient, scalable learning of stable and inductive entity embeddings for heterogeneous relational data. Created by Synerise.com team.

Python package for Bayesian Machine Learning with scikit-learn API

optimization routines for hyperparameter tuning

A scikit-learn-compatible Python implementation of ReBATE, a suite of Relief-based feature selection algorithms for Machine Learning.

scikit-learn wrappers for Python fastText.

⬛ Python Individual Conditional Expectation Plot Toolbox

InferPy: Deep Probabilistic Modeling with Tensorflow Made Easy

A library that implements fairness-aware machine learning algorithms

A library for hidden semi-Markov models with explicit durations