A scikit-learn-compatible library for estimating prediction intervals and controlling risks, based on conformal predictions.

Genetic Programming in Python, with a scikit-learn inspired API

Visual analysis and diagnostic tools to facilitate machine learning model selection.

A library of sklearn compatible categorical variable encoders

An optimization and data collection toolbox for convenient and fast prototyping of computationally expensive models.

Hummingbird compiles trained ML models into tensor computation for faster inference.

☁️ Build multimodal AI applications with cloud-native stack

Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.

Graph Algorithms

Chat with your documents on your local device using GPT models. No data leaves your device and 100% private.

PFRL: a PyTorch-based deep reinforcement learning library

DeltaPy - Tabular Data Augmentation (by @firmai)

Interactively explore unstructured datasets from your dataframe.

Graph Neural Network Library for PyTorch

Fast and Accurate ML in 3 Lines of Code

The Open Source Feature Store for AI/ML

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data