flax
abef9d81
- Merge pull request #5551 from mattjj:replace-lazy-init
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
4 days ago
Merge pull request #5551 from mattjj:replace-lazy-init PiperOrigin-RevId: 962826043
References
#5551 - replace lazy_init implementation to avoid jax partial_eval
Author
a-googler
Parents
3845cd94
f3b0e126
Loading
