jax.experimental.compilation_cache module

JAX disk compilation cache.

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

API

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