Utility
=======

.. automodule:: autopdex.utility
    :no-index:

.. autosummary::
   :toctree: _autosummary

   jit_with_docstring
   dict_zeros_like
   dict_flatten
   reshape_as
   mask_set
   mask_select
   mask_op
   search_neighborhood
   get_condition_number
   symmetry_check
   dof_select
   jnp_to_tuple
   to_jax_function
   jacfwd_upto_n_scalar_args
   jacfwd_upto_n_one_vector_arg

