jax.clear_caches

jax.clear_caches()

清除所有编译和分阶段缓存。

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