jax
f5ce76ba
- Merge pull request #19796 from mattjj:jax-attrs-jit-arg
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Merge pull request #19796 from mattjj:jax-attrs-jit-arg PiperOrigin-RevId: 606805982
References
#19796 - [attrs] allow passing a jax-attrs object to jit functions
Author
a-googler
Parents
cbe23b43
a45cc437
Loading