Skip to main content

A collection of PyTorch implementations of neural network architectures and layers.

Project description

Join Slack Twitter

LabML Neural Networks

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.


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



Recurrent Highway Networks


HyperNetworks - HyperLSTM

Capsule Networks

Generative Adversarial Networks

Sketch RNN

Reinforcement Learning


Normalization Layers


pip install labml-nn

Citing LabML

If you use LabML for academic research, please cite the library using the following BibTeX entry.

 author = {Varuna Jayasiri, Nipun Wijerathne},
 title = {LabML: A library to organize machine learning experiments},
 year = {2020},
 url = {},

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for labml-nn, version 0.4.89
Filename, size File type Python version Upload date Hashes
Filename, size labml_nn-0.4.89-py3-none-any.whl (154.9 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size labml-nn-0.4.89.tar.gz (109.5 kB) File type Source Python version None Upload date Hashes View

Supported by

Pingdom Pingdom Monitoring Google Google Object Storage and Download Analytics Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page