flax
15223579
- Merge pull request #346 from DanGraur:jax-utils-assert-message-fix
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
6 years ago
Merge pull request #346 from DanGraur:jax-utils-assert-message-fix PiperOrigin-RevId: 320826831
References
#346 - Added an error message to the assert in jax_utils.prefetch_to_device(*)
Author
a-googler
Parents
060737d4
2e7012da
Loading
