Skip to content

Commit 60a6cd4

Browse files
Add note on etils requirement for the Jax compilation cache.
The compilation cache has a dependency on etils.epath if the cache is not on a local filesystem. PiperOrigin-RevId: 694311585
1 parent 6e8a35f commit 60a6cd4

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

docs/persistent_compilation_cache.md

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,18 @@
11
# Persistent compilation cache
22

3-
<!--* freshness: { reviewed: '2024-04-09' } *-->
3+
<!--* freshness: { reviewed: '2024-11-07' } *-->
44

55
JAX has an optional disk cache for compiled programs. If enabled, JAX will
66
store copies of compiled programs on disk, which can save recompilation time
77
when running the same or similar tasks repeatedly.
88

9+
Note: if the compilation cache is not on a local filesystem,
10+
[etils](https://pypi.org/project/etils/) needs to be installed.
11+
12+
```python
13+
pip install etils
14+
```
15+
916
## Usage
1017

1118
### Quick start

0 commit comments

Comments
 (0)