jax authors b8b119d9b9 Cleanup deprecated compilation cache APIs. ...

Since the compilation cache is now initialized lazily,
existing APIs initialize_cache() and is_initialized()
are confusing. Deprecate these APIs.

Introduce a new API set_cache_dir() to explicitly set the
cache directory path in code.

Testing: revised unit tests, test workload.
PiperOrigin-RevId: 598073423

2024-01-12 22:44:48 -08:00

..

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

compilation_cache.py

Cleanup deprecated compilation cache APIs.

2024-01-12 22:44:48 -08:00