=================
API Documentation
=================

Batched maps
------------

.. autosummary::
    :toctree: _api/
    :recursive:

    adv_jax_math.batch_jacfwd
    adv_jax_math.batch_jacrev
    adv_jax_math.batch_map
    adv_jax_math.batch_vectorize
    adv_jax_math.batch_vmap

Sparse differentiation
----------------------

.. autosummary::
    :toctree: _api/
    :recursive:

    adv_jax_math.sparse_pullback
    adv_jax_math.sparse_pullback_map