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

JAX disk compilation cache.

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

API
---

.. autofunction:: set_cache_dir
.. autofunction:: reset_cache
