.. currentmodule:: stringjax_tools


stringjax_tools
===============

Package overview
----------------

``stringjax_tools`` is organised around small, composable JAX transform helpers.
The modules are intentionally independent from any StringJAX application
package.

.. toctree::
    :maxdepth: 1

    stringjax_tools.auto_vectorise
    stringjax_tools.vmap
    stringjax_tools.jit
    stringjax_tools.cache
    stringjax_tools.pytrees