jax.clear_caches

jax.clear_caches()

清除所有編譯和暫存快取。

這不會清除持久快取；若要停用它 (例如用於基準測試)，請將 jax_enable_compilation_cache 設定選項設為 False。