jax
6df7333b
- Fix hijax `VmapOf.batch_dim_rule` with `None` map dims.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
102 days ago
Fix hijax `VmapOf.batch_dim_rule` with `None` map dims. PiperOrigin-RevId: 875769536
References
#35385 - Fix hijax `VmapOf.batch_dim_rule` with `None` map dims.
Author
chr1sj0nes
Committer
Google-ML-Automation
Parents
7af1783e
Loading