hfai.nn
==========

.. toctree::
   :maxdepth: 2
   :caption: Contents:

.. py:currentmodule:: hfai.nn
.. autosummary::
   :nosignatures:
   
   to_hfai
   to_torch
   bench
   compare
   LSTM
   LSTM_fullc
   GRU
   LayerNorm
   MultiheadAttention
   Dropout
   Hardtanh
   LogSoftmax
   Softmax
   Softmax2d
   Softmin
   Softplus
   ReLU
   ReLU6
   Threshold
   LeakyReLU
   RReLU
   Hardsigmoid
   Hardshrink
   Softshrink


.. py:currentmodule:: hfai.nn
.. autofunction:: to_hfai
.. autofunction:: to_torch
.. autofunction:: bench
.. autofunction:: compare
.. autoclass:: LSTM
.. autoclass:: LSTM_fullc
.. autoclass:: GRU
.. autoclass:: LayerNorm
.. autoclass:: MultiheadAttention
.. autoclass:: Dropout
.. autoclass:: Hardtanh
.. autoclass:: LogSoftmax
.. autoclass:: Softmax
.. autoclass:: Softmax2d
.. autoclass:: Softmin
.. autoclass:: Softplus
.. autoclass:: ReLU
.. autoclass:: ReLU6
.. autoclass:: Threshold
.. autoclass:: LeakyReLU
.. autoclass:: RReLU
.. autoclass:: Hardsigmoid
.. autoclass:: Hardshrink
.. autoclass:: Softshrink