StringJAX Tools#
StringJAX Tools is a small support package for reusable JAX transform
patterns used across StringJAX projects. It collects conservative
auto-vectorisation, cached manual vmap/jit wrappers, static-argument
JIT helpers, explicit compilation-cache setup, and configurable pytree
registration utilities.
The package is intentionally application-agnostic. It should not know about flux vacua, Calabi-Yau moduli, charge conventions, or any other package-level domain language. StringJAX packages provide their own naming and state policies; this package provides the JAX mechanics.
Recommended first path#
Read Usage Guide for the intended design and migration style.
Run through Tutorial Notebooks for executable feature examples.
Use stringjax_tools for the API reference.
Add package-local policies, such as pytree static and ignored keys, in the consuming package rather than in
stringjax_tools.