jax
89c404e7 - Improve error message when a global jax.Array is closed over a jitted function in McJAX.

Commit
2 years ago
Improve error message when a global jax.Array is closed over a jitted function in McJAX. PiperOrigin-RevId: 648010704
Author
Committer
Parents
Loading