A collection of PyTorch implementations of neural network architectures and layers.
Project description
labml.ai 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. for updates.
Modules
✨ Transformers
- Multi-headed attention
- Transformer building blocks
- Transformer XL
- Compressive Transformer
- GPT Architecture
- GLU Variants
- kNN-LM: Generalization through Memorization
- Feedback Transformer
- Switch Transformer
- Fast Weights Transformer
✨ Recurrent Highway Networks
✨ LSTM
✨ HyperNetworks - HyperLSTM
✨ Capsule Networks
✨ Generative Adversarial Networks
✨ Sketch RNN
✨ Reinforcement Learning
- Proximal Policy Optimization with Generalized Advantage Estimation
- Deep Q Networks with with Dueling Network, Prioritized Replay and Double Q Network.
✨ Optimizers
✨ Normalization Layers
Installation
pip install labml-nn
Citing LabML
If you use LabML for academic research, please cite the library using the following BibTeX entry.
@misc{labml,
author = {Varuna Jayasiri, Nipun Wijerathne},
title = {LabML: A library to organize machine learning experiments},
year = {2020},
url = {https://nn.labml.ai/},
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
labml_nn-0.4.92-py3-none-any.whl
(171.7 kB
view hashes)
Close
Hashes for labml_nn-0.4.92-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a1aa60e6365c161a315c0623da8c2cb2bf0316914cb4cd6b56b0fda90e3e7d85 |
|
MD5 | 8ad0e944bdcf06f26b3a72610702f401 |
|
BLAKE2b-256 | 58617056675e513a373df0d76abd49f1e752c425bc53bb870934216744881f79 |