wrap
====

.. currentmodule:: penzai.core.named_axes

.. autofunction:: wrap