annotated_deep_learning_paper_implementations README

The project's README from github.com/labmlai/annotated_deep_learning_paper_implementations. As one markdown file: README.md.

Twitter

labml.ai Deep Learning Paper Implementations

This is a collection of simple PyTorch implementations of neural networks and related algorithms. These implementations are documented with explanations,

The website renders these as side-by-side formatted notes. We believe these would help you understand these algorithms better.

Screenshot

We are actively maintaining this repo and adding new implementations almost weekly. Twitter for updates.

Paper Implementations

✨ Transformers

✨ Low-Rank Adaptation (LoRA)

✨ Eleuther GPT-NeoX

✨ Diffusion models

✨ Generative Adversarial Networks

✨ Recurrent Highway Networks

✨ LSTM

✨ HyperNetworks - HyperLSTM

✨ ResNet

✨ ConvMixer

✨ Capsule Networks

✨ U-Net

✨ Sketch RNN

✨ Graph Neural Networks

✨ Counterfactual Regret Minimization (CFR)

Solving games with incomplete information such as poker with CFR.

✨ Reinforcement Learning

✨ Optimizers

✨ Normalization Layers

✨ Distillation

✨ Adaptive Computation

✨ Uncertainty

✨ Activations

✨ Langauge Model Sampling Techniques

✨ Scalable Training/Inference

Installation

pip install labml-nn

annotated_deep_learning_paper_implementations (labmlai/annotated_deep_learning_paper_implementations)

Searching Database…

What is this?A collection of 60+ PyTorch implementations of deep learning papers, each annotated with side-by-side explanatory notes so you can read the code and the math/explanation together.
Is this a library I import, or more like a reading resource?It's primarily a literate-programming reading resource: annotated source code meant to be read and studied rather than a packaged library for production use, though you can still run or copy the implementations.
How do I actually read or run these implementations?Browse the annotated notes on the labml.ai website for side-by-side code and explanations, or clone the repo and run the individual Python/PyTorch scripts directly.
What do I need to run the code?Python and PyTorch, since every implementation is written as a simple PyTorch script.
Are these production-ready implementations?No — they're deliberately kept simple and explanatory ('simple PyTorch implementations') to aid understanding, not optimized or hardened for production use.
What topics are covered besides transformers?GANs (CycleGAN, StyleGAN2, Wasserstein GAN), diffusion models (DDPM, DDIM, Stable Diffusion), reinforcement learning (PPO, DQN), graph neural networks, capsule networks, optimizers (Adam, AdaBelief, Sophia), normalization layers, and more.
Is the project actively maintained?Yes, the maintainers say they add new implementations almost weekly.
Does it cover LoRA or large language model techniques?Yes, it includes a Low-Rank Adaptation (LoRA) implementation and GPT-NeoX examples covering generation, fine-tuning, and LLM.int8() quantization.
What is special about the way the code is presented?Each implementation uses literate programming style, pairing code line-by-line with explanatory notes rendered as a side-by-side document on the labml.ai website.
Creator
labmlai
Size
152.7 MB
Stars
67.6K
Updated
CommitJan 22
Age
6y
Commits · 90 Days