API#

jax.experimental.compilation_cache.compilation_cache.set_cache_dir(path)[source]#

Sets the persistent compilation cache directory.

After calling this, jit-compiled functions are saved topath, so theydo not need be recompiled if the process is restarted or otherwise run again.This also tells Jax where to look for compiled functions before compiling.

Return type:

None

jax.experimental.compilation_cache.compilation_cache.reset_cache()[source]#

Get back to pristine, uninitialized state.

Return type:

None