Skip to main content

> ML_LIBRARIES_CATALOG_v1.0

Qualified ML Libraries

Every library is independently qualified with primary citations, supported version ranges, hardware accelerator separation, and real-world failure patterns.

Showing 18 of 161 Qualified Libraries & Tools (Page 7 of 9)Accelerator Mappings & Failure Modes
automl-hpo
3.6.1MIT

Optuna

Preferred Networks

The industry-standard Bayesian hyperparameter optimization framework with define-by-run API.

Training:
CPUCUDA
Inference:
#tabular#text#image#audio
automl-hpo
0.2.7BSD-3-Clause

Hyperopt

Hyperopt Community

Pioneering Bayesian hyperparameter optimization library implementing Tree of Parzen Estimators.

Training:
CPU
Inference:
#tabular
automl-hpo
0.15.0BSD-3-Clause

auto-sklearn

AutoML Freiburg

Groundbreaking academic automated machine learning framework built on scikit-learn and SMAC.

Training:
CPU
Inference:
CPU
#tabular
automl-hpo
0.12.2LGPL-3.0

TPOT

Epistasis Lab / University of Pennsylvania

Genetic programming automated machine learning tool that exports clean, human-readable Python pipelines.

Training:
CPU
Inference:
CPU
#tabular
automl-hpo
0.10.1Apache-2.0

Ludwig

Linux Foundation AI & Data / Predibase / Uber

Declarative, low-code deep learning framework for multimodal AI and LLM fine-tuning via YAML configs.

Training:
CPUCUDAROCMMPS
Inference:
CPUCUDAROCMMPS
#tabular#text#image#audio
privacy-security-optimization
0.46.0MIT

SHAP

Scott Lundberg / SHAP Community

The industry-standard game-theoretic machine learning explainability and feature attribution framework.

Training:
Not Supported (Inference Only)
Inference:
CPUCUDA
#tabular#text#image
privacy-security-optimization
0.2.0.1BSD-2-Clause

LIME

Marco Tulio Ribeiro / University of Washington

Local Interpretable Model-agnostic Explanations for explaining individual black-box predictions.

Training:
Not Supported (Inference Only)
Inference:
CPU
#tabular#text#image
privacy-security-optimization
0.7.0BSD-3-Clause

Captum

PyTorch Foundation / Meta

Official PyTorch model interpretability library implementing gradient-based attribution algorithms.

Training:
Not Supported (Inference Only)
Inference:
CPUCUDAROCMMPS
#text#image#tabular
privacy-security-optimization
0.6.2MIT

InterpretML

Microsoft Research

Microsoft Research toolkit for training glassbox Explainable Boosting Machines and explaining black-box AI.

Training:
CPU
Inference:
CPU
#tabular
privacy-security-optimization
0.11.0MIT

Fairlearn

Fairlearn Community / NumFOCUS / Microsoft

Community-driven Python library for auditing algorithmic fairness and mitigating demographic disparities.

Training:
CPU
Inference:
CPU
#tabular
privacy-security-optimization
0.6.1Apache-2.0

AI Fairness 360 (AIF360)

Linux Foundation AI & Data / IBM Research

IBM and Linux Foundation AI toolkit providing over 70 fairness metrics and 10 bias mitigation algorithms.

Training:
CPU
Inference:
CPU
#tabular
privacy-security-optimization
0.9.6Apache-2.0

Alibi

Seldon Technologies

Seldon's model inspection library specializing in counterfactual recourse and anchor explanations.

Training:
Not Supported (Inference Only)
Inference:
CPUCUDA
#tabular#text#image
Observability & Drift
0.4.38Apache-2.0

Evidently AI

Evidently AI Inc.

Open-source evaluation, testing, and observability framework for machine learning and LLM applications.

Training:
Not Supported (Inference Only)
Inference:
CPU
#tabular#text
Observability & Drift
0.18.1Apache-2.0

Deepchecks

Deepchecks

Holistic testing and validation framework for tabular data, computer vision, and LLM applications.

Training:
Not Supported (Inference Only)
Inference:
CPU
#tabular#image#text
Observability & Drift
1.4.11Apache-2.0

whylogs

WhyLabs

Lightweight, privacy-preserving statistical data profiling library powered by Apache DataSketches.

Training:
Not Supported (Inference Only)
Inference:
CPU
#tabular#text
Observability & Drift
1.0.3Apache-2.0

Great Expectations

Superconductive / Great Expectations Community

The industry-standard open-source framework for declarative data pipeline testing and automated validation.

Training:
Not Supported (Inference Only)
Inference:
CPU
#tabular
Observability & Drift
4.15.0ELv2 (Permissive for internal enterprise use)

Arize Phoenix

Arize AI

Open-source AI observability, LLM tracing, and RAG evaluation platform built on OpenTelemetry.

Training:
Not Supported (Inference Only)
Inference:
CPU
#text
Observability & Drift
0.1.18Apache-2.0

Ragas

Exploding Gradients

Automated evaluation framework for Retrieval Augmented Generation (RAG) pipelines.

Training:
Not Supported (Inference Only)
Inference:
CPU
#text