diff --git a/docs/performance-tips.rst b/docs/performance-tips.rst index f4f7c0a3a..5206ff941 100644 --- a/docs/performance-tips.rst +++ b/docs/performance-tips.rst @@ -14,7 +14,7 @@ Although the compiled code is fast, it still takes time to compile. If you are r jax.config.update("jax_persistent_cache_min_entry_size_bytes", -1) jax.config.update("jax_persistent_cache_min_compile_time_secs", 0) -This will create a cache directory called ``jax-caches`` in the parent directory of the script. The ``jax_persistent_cache_min_entry_size_bytes`` and ``jax_persistent_cache_min_compile_time_secs`` parameters are set to -1 and 0, respectively, to ensure that all compiled code is cached. For more details on caching, refer to official JAX documentation https://jax.readthedocs.io/en/latest/persistent_compilation_cache.html#persistent-compilation-cache. +This will create a cache directory called ``jax-caches`` in the parent directory of the script. The ``jax_persistent_cache_min_entry_size_bytes`` and ``jax_persistent_cache_min_compile_time_secs`` parameters are set to -1 and 0, respectively, to ensure that all compiled code is cached. For more details on caching, refer to official JAX documentation `here `__. Note: Updating JAX version might re-compile some previously cached code, and thi might increase the cache size. Every once in a while, you might need to clear your cache directory.