jax.clear_caches

jax.clear_caches()

Clear all compilation and staging caches.

This doesn’t clear the persistent cache; to disable it (e.g. for benchmarks), set the jax_enable_compilation_cache config option to False.