mirror of
https://github.com/labmlai/annotated_deep_learning_paper_implementations.git
synced 2025-10-29 01:26:44 +08:00
53 lines
1.4 KiB
ReStructuredText
53 lines
1.4 KiB
ReStructuredText
.. image:: https://badge.fury.io/py/labml-nn.svg
|
|
:target: https://badge.fury.io/py/labml-nn
|
|
.. image:: https://pepy.tech/badge/labml-nn
|
|
:target: https://pepy.tech/project/labml-nn
|
|
|
|
LabML Neural Networks
|
|
=====================
|
|
|
|
This is a collection of simple PyTorch implementation of various neural network architectures and layers. We will keep adding to this.
|
|
|
|
Transformers
|
|
------------
|
|
|
|
`Transformers module <http://lab-ml.com/labml_nn/transformers/__init__>`_ contains implementations for
|
|
`multi-headed attention <http://lab-ml.com/labml_nn/transformers/mha.html>`_
|
|
and
|
|
`relative multi-headed attention <http://lab-ml.com/labml_nn/transformers/relative_mha.html>`_.
|
|
|
|
|
|
Installation
|
|
------------
|
|
|
|
.. code-block:: console
|
|
|
|
pip install labml_nn
|
|
|
|
Links
|
|
-----
|
|
|
|
`💬 Slack workspace for discussions <https://join.slack.com/t/labforml/shared_invite/zt-egj9zvq9-Dl3hhZqobexgT7aVKnD14g/>`_
|
|
|
|
`📗 Documentation <http://lab-ml.com/>`_
|
|
|
|
`📑 Articles & Tutorials <https://medium.com/@labml/>`_
|
|
|
|
`👨🏫 Samples <https://github.com/lab-ml/samples>`_
|
|
|
|
|
|
Citing LabML
|
|
------------
|
|
|
|
If you use LabML for academic research, please cite the library using the following BibTeX entry.
|
|
|
|
.. code-block:: bibtex
|
|
|
|
@misc{labml,
|
|
author = {Varuna Jayasiri, Nipun Wijerathne},
|
|
title = {LabML: A library to organize machine learning experiments},
|
|
year = {2020},
|
|
url = {https://lab-ml.com/},
|
|
}
|
|
|