layer\_stack
============

.. automodule:: penzai.nn.layer_stack

   
   
   .. rubric:: Classes

   .. autosummary::
      :toctree: leaf
      :template: pzclass.rst
   
      LayerStack
      LayerStackGetAttrKey
      LayerStackVarBehavior
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
      :toctree: leaf
      :template: pzbase.rst
   
      layerstack_axes_from_keypath