stringjax_tools.auto_vectorise
==============================

.. currentmodule:: stringjax_tools.auto_vectorise

.. automodule:: stringjax_tools.auto_vectorise


Core decorator
--------------

.. autosummary::
    :toctree: _autosummary
    :template: custom-class-template.rst

    ArgSpec

.. autosummary::
    :toctree: _autosummary

    auto_vmap


Defaults and cache management
-----------------------------

.. autosummary::
    :toctree: _autosummary

    set_auto_vmap_defaults
    get_auto_vmap_defaults
    reset_auto_vmap_defaults
    auto_vmap_defaults
    set_auto_vmap_default_shapes
    get_auto_vmap_default_shapes
    reset_auto_vmap_default_shapes
    auto_vmap_default_shapes
    clear_auto_vmap_caches