jax_convenience_fns
-----------------------------

.. currentmodule:: luas.jax_convenience_fns

.. automodule:: luas.jax_convenience_fns

.. autosummary::
   :toctree: summary
   
   array_to_pytree_2D
   pytree_to_array_2D
   varying_params_wrapper
   large_hessian_calc
   transf_from_unbounded_params
   transf_to_unbounded_params