A collection of PyTorch implementations of neural network architectures and layers.
Project description
LabML Neural Networks
This is a collection of simple PyTorch implementation of various neural network architectures and layers. We will keep adding to this.
Modules
✨ Transformers
Transformers module contains implementations for multi-headed attention and relative multi-headed attention.
✨ Recurrent Highway Networks
✨ LSTM
✨ 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.
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://lab-ml.com/},
}
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 Distribution
labml_nn-0.4.63.tar.gz
(136.2 kB
view hashes)
Built Distribution
labml_nn-0.4.63-py3-none-any.whl
(202.2 kB
view hashes)
Close
Hashes for labml_nn-0.4.63-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ca73fa4e432c3955f9f2bce5d27e5722f6754ba6dd9c8f4461cbb25cced5a823 |
|
MD5 | 88f466cdcb6139511533126b9ab11c78 |
|
BLAKE2b-256 | c81baee2543416c214479e0b1c01f2583f55f01e9c49b7f6315fdc2f955968cf |