pyepo.func.jax.utils
====================

.. py:module:: pyepo.func.jax.utils

.. autoapi-nested-parse::

   Utility functions for the JAX frontend