flax
e4f417ac
- [nnx] vectorize vmap split counts
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[nnx] vectorize vmap split counts
References
#3989 - [nnx] vectorize vmap split counts
Author
cgarciae
Committer
cgarciae
Parents
998792a4
Loading