Skip to content

Commit

Permalink
fix the url link
Browse files Browse the repository at this point in the history
  • Loading branch information
YigitElma committed Dec 20, 2024
1 parent a6179ac commit 2ab9f0d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/performance-tips.rst
Original file line number Diff line number Diff line change
Expand Up @@ -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 <https://jax.readthedocs.io/en/latest/persistent_compilation_cache.html#persistent-compilation-cache>`__.

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.

Expand Down

0 comments on commit 2ab9f0d

Please sign in to comment.