Stable Baselines3 Contrib alternatives for AI agents.

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

Superfast AI decision making and intelligent processing of multi-modal data.

Open-source evaluation toolkit of large multi-modality models (LMMs), support 220+ LMMs, 80+ benchmarks

AI watermark remover. CLI and Python library to strip visible and invisible AI watermarks (Gemini / Nano Banana sparkle, SynthID) and provenance metadata (C2PA, EXIF, IPTC) from images.

AI IDE for hardware development, support Arduino, MicroPython, ESP32, STM32, RP2040, Nrf5x...

The Compute Library is a set of computer vision and machine learning functions optimised for both Arm CPUs and GPUs using SIMD technologies.

3D Computer Vision Framework

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

Deep Learning for humans

Ultralytics YOLOv5 in PyTorch > ONNX > CoreML > TFLite

Deep universal probabilistic programming with Python and PyTorch

Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.

The fastai deep learning library

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

The Unity Machine Learning Agents Toolkit (ML-Agents) is an open-source project that enables games and simulations to serve as environments for training intelligent agents using deep reinforcement learning and imitation learning.

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.