Shapegan 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

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

Deep universal probabilistic programming with Python and PyTorch

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

The fastai deep learning library

Ultralytics YOLOv5 in PyTorch > ONNX > CoreML > TFLite

🧑‍🏫 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, ... 🧠

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Datasets, Transforms and Models specific to Computer Vision

NLTK Source

A hyperparameter optimization framework

ChatterBot is a machine learning, conversational dialog engine for creating chat bots

The Triton Inference Server provides an optimized cloud and edge inferencing solution.

Ultralytics YOLOv3 in PyTorch > ONNX > CoreML > TFLite

🐍 Geometric Computer Vision Library for Spatial AI