jax
de6ce2a5
- allow vmap in_axes to be lists
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
6 years ago
allow vmap in_axes to be lists
References
#1235 - allow vmap in_axes to be lists
Author
mattjj
Parents
0c832b2d
Loading
