Stable Baselines3 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

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

Deep universal probabilistic programming with Python and PyTorch

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

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

tensorboard for pytorch (and chainer, mxnet, numpy, ...)

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

The fastai deep learning library

Hummingbird compiles trained ML models into tensor computation for faster inference.

Classic papers and resources on recommendation

Ultralytics YOLOv5 in PyTorch > ONNX > CoreML > TFLite

StarCraft II Learning Environment

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Datasets, Transforms and Models specific to Computer Vision