``jax.experimental.compilation_cache`` module
===============================================

JAX disk compilation cache.

.. automodule:: jax.experimental.compilation_cache.compilation_cache

API
---

.. autofunction:: is_initialized
.. autofunction:: initialize_cache
.. autofunction:: set_cache_dir
.. autofunction:: reset_cache