Awesome Machine Learning

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

🧠 Make your agents learn from experience. Now available as a hosted solution at kayba.ai

An intuitive library to add plotting functionality to scikit-learn objects.

A modular active learning framework for Python

Karate Club: An API Oriented Open-source Python Framework for Unsupervised Learning on Graphs (CIKM 2020)

Feature engineering and selection open-source Python library compatible with sklearn.

A toolkit for reproducible reinforcement learning research.

Open-source persistent memory for AI agent pipelines (LangGraph, CrewAI, AutoGen) and Claude. REST API + knowledge graph + autonomous consolidation.

Automatically Visualize any dataset, any size with a single line of code. Created by Ram Seshadri. Collaborators Welcome. Permission Granted upon Request.

Hyper-parameter optimization for sklearn

Machine learning with dataframes

The Python ensemble sampling toolkit for affine-invariant MCMC

Spearmint Bayesian optimization codebase

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

A security scanner for your LLM agentic workflows

The Classical Language Toolkit

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

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

Little Ball of Fur - A graph sampling extension library for NetworKit and NetworkX (CIKM 2020)

Pandas, Polars, Spark, and Snowpark DataFrame comparison for humans and more!

No description.

A set of tools for creating and testing machine learning features, with a scikit-learn compatible API

Hyperparameter optimization and feature selection for scikit-learn using evolutionary algorithms. A modern alternative to GridSearchCV and RandomizedSearchCV.

The official implementation of "The Shapley Value of Classifiers in Ensemble Games" (CIKM 2021).

Library for machine learning stacking generalization.

Build, test, deploy, iterate - Dev and prod tool for data science pipelines