Skip to main content

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

Project description

PiPy Version PiPy Downloads

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

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/},
}

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.62.tar.gz (115.7 kB view hashes)

Uploaded Source

Built Distribution

labml_nn-0.4.62-py3-none-any.whl (167.7 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page