Enable JAX persistent caching via `EconPizzaConfig` #17

shtopane · 2024-10-10T14:00:12Z

Expose a configuration object to the end user with the option to enable persistent caching via JAX.
By default, the folder where the JAX files reside is called __jax_cache__ and is created in the user's working directory. This can be changed by the option jax_cache_folder.

Usage:

import econpizza as ep
ep.config.update("enable_jax_persistent_cache", True)

Optionally, change the name or destination of the caching folder:

import econpizza as ep
ep.config.update("jax_cache_folder", "example_folder")
ep.config.update("enable_jax_persistent_cache", True)

gboehl · 2024-10-11T11:13:56Z

Thank you!

Enable JAX persistent caching via EconPizzaConfig

c804b0c

gboehl merged commit 958d700 into gboehl:master Oct 11, 2024
9 checks passed

Provide feedback