jax_to_numpy
============

.. currentmodule:: muse.utils

.. autofunction:: jax_to_numpy