Is there a way to share the JIT cache between processes

[alonkukl]

In my scenario, I have a couple of workers (processes) running on the same machine.
The workers run JAX and could benefit from sharing the JIT cache, is there a way to accomplish it?

[jakevdp]

I don't know of any way of sharing the in-memory cache between processes, but there is experimental support for on-disk peristent compilation cache It currently only supports TPU, but a GPU implementation is currently in the works.