``brainstate.nn`` module
========================

.. currentmodule:: brainstate.nn
.. automodule:: brainstate.nn

Neural network building blocks for constructing and training artificial and
spiking neural networks: the core module system, parameter containers and
transforms, a full library of layers, dynamical-systems primitives, and
training utilities. Browse the reference by category below.

.. toctree::
   :maxdepth: 1

   module
   parameters
   regularization
   linear
   conv
   normalization
   pooling
   padding
   dropout
   activation
   embedding
   connectivity
   rnn
   dynamics
   metrics
   utils