MaxViT alternatives for AI agents.

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

Deep Learning for humans

Ultralytics YOLO 🚀

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

🐍 Geometric Computer Vision Library for Spatial AI

Deep universal probabilistic programming with Python and PyTorch

The fastai deep learning library

A toolkit for making real world machine learning and data analysis applications in C++

Ultralytics YOLOv5 in PyTorch > ONNX > CoreML > TFLite

We write your reusable computer vision tools. 💜

🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠

🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools

Tensors and Dynamic neural networks in Python with strong GPU acceleration

A collection of tutorials on state-of-the-art computer vision models and techniques. Explore everything from foundational architectures like ResNet to cutting-edge models like RF-DETR, YOLO11, SAM 3, and Qwen3-VL.

Datasets, Transforms and Models specific to Computer Vision

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