nerva\_jax.utilities
====================

.. automodule:: nerva_jax.utilities

   
   .. rubric:: Functions

   .. autosummary::
   
      disable_gpu
      load_dict_from_npz
      parse_function_call
      pp
      pp_numpy
      save_dict_to_npz
      set_jax_options
   
   .. rubric:: Classes

   .. autosummary::
   
      FunctionCall
      StopWatch