numpy_to_jax
============

.. currentmodule:: muse.utils

.. autofunction:: numpy_to_jax