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

References

Author

mattjj

Committer

mattjj

Parents