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 4 of 9)Accelerator Mappings & Failure Modes
Inference Serving
1.19.2MIT

ONNX Runtime

Microsoft / Linux Foundation AI & Data

Cross-platform, high-performance ML inferencing and training accelerator.

Training:
CPUCUDAROCM
Inference:
CPUCUDAROCMMPSWEBGPUWASM
#tabular#text#image#audio#multimodal
Inference Serving
2.49.0BSD-3-Clause

Triton Inference Server

NVIDIA

Enterprise inference serving software for high-throughput, multi-framework model deployment.

Training:
Not Supported (Inference Only)
Inference:
CPUCUDAROCM
#tabular#text#image#audio#multimodal
Inference Serving
2024.4.0Apache-2.0

OpenVINO

Intel

Open-source toolkit for optimizing and deploying AI inference on Intel hardware.

Training:
Not Supported (Inference Only)
Inference:
CPU
#image#text#audio#multimodal
Inference Serving
10.4.0NVIDIA Software License

TensorRT

NVIDIA

High-performance deep learning inference optimizer and runtime by NVIDIA.

Training:
Not Supported (Inference Only)
Inference:
CUDA
#image#text#audio
Inference Serving
0.17.0Apache-2.0

Apache TVM

Apache Software Foundation

An end-to-end deep learning compiler for CPUs, GPUs and specialized accelerators.

Training:
Not Supported (Inference Only)
Inference:
CPUCUDAROCMMPSWEBGPUWASM
#tabular#text#image#audio
client-edge-embedded
0.3.0BSD-3-Clause

ExecuTorch

Meta / PyTorch Foundation

End-to-end solution for enabling on-device AI across mobile and edge devices for PyTorch.

Training:
Not Supported (Inference Only)
Inference:
CPUMPS
#image#text#audio
client-edge-embedded
2.17.0Apache-2.0

TensorFlow Lite

Google

Deploy machine learning models on mobile and edge devices.

Training:
Not Supported (Inference Only)
Inference:
CPUMPS
#image#audio#text
client-edge-embedded
Core ML 8 (macOS 15 / iOS 18)Proprietary Apple Framework

Core ML

Apple

Integrate machine learning models into your Apple app with hardware acceleration.

Training:
Not Supported (Inference Only)
Inference:
CPUMPS
#image#text#audio#multimodal
client-edge-embedded
0.2.62Apache-2.0

WebLLM

MLC AI

High-performance in-browser LLM inference engine powered by WebGPU.

Training:
Not Supported (Inference Only)
Inference:
WEBGPUWASM
#text
computer-vision
0.24.0BSD-3-Clause

scikit-image

scikit-image Development Team / NumFOCUS

Peer-reviewed scientific image processing algorithms built natively on NumPy.

Training:
Not Supported (Inference Only)
Inference:
CPU
#image
computer-vision
4.10.0Apache-2.0

OpenCV

OpenCV Foundation / Open Source Vision Foundation

The foundational open-source computer vision and video processing library.

Training:
CPUCUDAXPU
Inference:
CPUCUDAWEBGPUWASM
#image#video
computer-vision
0.19.1BSD-3-Clause

torchvision

PyTorch Foundation / Meta

Official PyTorch library for computer vision datasets, architectures, and transforms.

Training:
CPUCUDAROCMMPSXPUTPU
Inference:
CPUCUDAROCMMPS
#image#video
computer-vision
1.0.9Apache-2.0

timm (PyTorch Image Models)

Ross Wightman / Hugging Face

The definitive collection of PyTorch vision architectures, pretrained weights, and training recipes.

Training:
CPUCUDAROCMMPSXPUTPU
Inference:
CPUCUDAROCMMPS
#image
computer-vision
8.3.0AGPL-3.0 (Commercial license required for proprietary distribution)

Ultralytics (YOLO)

Ultralytics Inc.

The industry-standard framework for real-time object detection, segmentation, and pose tracking.

Training:
CPUCUDAROCMMPSXPU
Inference:
CPUCUDAROCMMPSWEBGPUWASM
#image#video
computer-vision
0.6Apache-2.0

Detectron2

Meta FAIR (Fundamental AI Research)

Meta FAIR's modular deep learning platform for object detection and instance segmentation.

Training:
CPUCUDA
Inference:
CPUCUDA
#image
computer-vision
3.3.0Apache-2.0

MMDetection

OpenMMLab

OpenMMLab's comprehensive toolbox implementing over 70 object detection and segmentation architectures.

Training:
CPUCUDA
Inference:
CPUCUDA
#image#video
computer-vision
1.4.15MIT

Albumentations

Albumentations Team

High-performance image augmentation library for deep learning computer vision.

Training:
CPU
Inference:
#image
computer-vision
0.7.3Apache-2.0

Kornia

Kornia AI / PyTorch Ecosystem

Differentiable computer vision library built on PyTorch for GPU-accelerated operators.

Training:
CPUCUDAROCMMPSXPUTPU
Inference:
CPUCUDAROCMMPS
#image