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

.. currentmodule:: penzai.nn.layer_stack

.. autofunction:: layerstack_axes_from_keypath