Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Enable JAX persistent caching via EconPizzaConfig class #16

Closed
wants to merge 1 commit into from

Conversation

shtopane
Copy link
Contributor

  • 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
Copy link
Owner

gboehl commented Oct 10, 2024

Perfect! Could you cleanup the commit history a bit? It should be sufficient to have all this as one commit. I'd use git rebase interactively for that.

@shtopane
Copy link
Contributor Author

Okay, something messed up, I'll close this one and open another

@shtopane shtopane closed this Oct 10, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants