Pipeline parallelism for the minimalist

Nano vLLM

A JAX research toolkit for building, editing, and visualizing neural networks.

This repo implements a Stable Diffusion model in PyTorch with all the essential components.

Measuring Massive Multitask Language Understanding | ICLR 2021

JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

TFRecorder makes it easy to create TensorFlow records (TFRecords) from Pandas DataFrames and CSVs files containing images or structured data.

FinRL®: Financial Reinforcement Learning. 🔥

Taming Transformers for High-Resolution Image Synthesis

Trax — Deep Learning with Clear Code and Speed

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

Ultimate Python study guide 🐍 🐍 🐍

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

A small package to create visualizations of PyTorch execution graphs

Neural Turing Machines (NTM) - PyTorch Implementation

An easy-to-follow Pytorch implementation of Hinton's Capsule Network

PyTorch tutorials.

Kubernetes IN Docker - local clusters for testing Kubernetes

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

A GPipe implementation in PyTorch

Open Source Neural Machine Translation and (Large) Language Models in PyTorch

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Automated provisioner of a Google Cloud TPU environment for training in PyTorch

Advanced Deep Learning and Reinforcement Learning course taught at UCL in partnership with Deepmind

Enabling PyTorch on XLA Devices (e.g. Google TPU)

Google Brain AutoML