flax
0d2251aa
- Add axis_name arg to lifted vmap
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
4 years ago
Add axis_name arg to lifted vmap
References
#1024 - Add axis_name arg to lifted vmap
Author
jheek
Parents
d5a7fe1d
Loading
