flax
0d2251aa - Add axis_name arg to lifted vmap
Commit
View On GitHub
Commit
5 years ago
Add axis_name arg to lifted vmap
Author
jheek jheek
Parents
d5a7fe1d
FAQ Terms Privacy Refunds Impressum
Loading