jax.clear_caches

jax.clear_caches()

清除所有编译和暂存缓存。

这不会清除持久缓存；要禁用它（例如，用于基准测试），请将 `jax_enable_compilation_cache` 配置选项设置为 `False`。