Fix cache init when JAX Array is created early (#25768)#25889

Open

Stella-S-Yan wants to merge 1 commit intojax-ml:mainfrom Stella-S-Yan:cache_reset

+48-5

Commits on Jan 22, 2025

Fix cache init when JAX Array is created early (jax-ml#25768 )
Stella-S-Yan
committed