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

Datasets, Transforms and Models specific to Computer Vision

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

NLTK Source

A hyperparameter optimization framework

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

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

Uniform Manifold Approximation and Projection

A self-hosted open source photo management service.

⚡ TabPFN: Foundation Model for Tabular Data ⚡

Deep Learning for humans

🧑‍🏫 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

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

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

Ultralytics YOLOv3 in PyTorch > ONNX > CoreML > TFLite