Tutorial Notebooks#
The tutorial notebooks are short executable examples for the main helper
families in stringjax_tools. They are intended as API orientation rather
than performance benchmarks.
Recommended path#
Start with automatic vectorisation if you want a function to accept either one sample or a paired leading-axis batch.
Use the manual vmap helpers when the batching axes are already explicit.
Read the JIT and cache notebooks when tuning compilation behavior.
Use the pytree notebook when registering stateful model classes.