jax

88dd29a0 - Re-enable persistent cache on cpu.

Re-enable persistent cache on cpu. CPU cache key now includes machine attributes, so there should no longer be a problem with incompatible CPUs accessing the same cache entry. PiperOrigin-RevId: 621341638