stringjax_tools.auto_vectorise.clear_auto_vmap_caches

clear_auto_vmap_caches(clear_jax=True)

Clear cached wrappers created by auto_vmap().

Parameters:

clear_jax (bool) – If True, also call jax.clear_caches() when available. This clears JAX’s in-process compilation caches but does not remove entries from a persistent on-disk compilation cache.

Return type:

None