penzai.experimental.v2.nn
=========================

.. automodule:: penzai.experimental.v2.nn

   
   
   

   
   
   

   
   
   

   
   
   



.. rubric:: Submodules

.. autosummary::
   :toctree:
   :template: pzmodule.rst
   :recursive:

   penzai.experimental.v2.nn.attention
   penzai.experimental.v2.nn.basic_ops
   penzai.experimental.v2.nn.combinators
   penzai.experimental.v2.nn.dropout
   penzai.experimental.v2.nn.embeddings
   penzai.experimental.v2.nn.grouping
   penzai.experimental.v2.nn.layer
   penzai.experimental.v2.nn.layer_stack
   penzai.experimental.v2.nn.linear_and_affine
   penzai.experimental.v2.nn.parameters
   penzai.experimental.v2.nn.standardization