layerstack\_axes\_from\_keypath
===============================

.. currentmodule:: penzai.experimental.v2.nn.layer_stack

.. autofunction:: layerstack_axes_from_keypath