stringjax_tools.cache

stringjax_tools.cache#

Opt-in helpers for JAX compilation-cache configuration.

configure_compilation_cache([cache_dir, ...])

Configure JAX's persistent compilation cache explicitly.