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

Deep Learning for humans

🐍 Geometric Computer Vision Library for Spatial AI

A python library for user-friendly forecasting and anomaly detection on time series.

Deep universal probabilistic programming with Python and PyTorch

Accessible large language models via k-bit quantization for PyTorch.

A self-hosted open source photo management service.

The fastai deep learning library

🤗 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.

Ultralytics YOLOv5 in PyTorch > ONNX > CoreML > TFLite

We write your reusable computer vision tools. 💜

The world's simplest facial recognition api for Python and the command line

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Datasets, Transforms and Models specific to Computer Vision

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

NLTK Source