flax
66544200
- Fix nits
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
110 days ago
Fix nits
References
#5102 - Add out_sharding argument to call methods for layers with jax calls that support it
Author
samanklesaria
Parents
1f61d6b7
Loading
