jax
a45cc437 - [attrs] allow passing a jax-attrs object to jit functions

Commit
1 year ago
[attrs] allow passing a jax-attrs object to jit functions currently we don't get any interesting cache hits; only on object identity match
Author
Committer
Parents
Loading