flax
afc169eb
- Update tests/jax_utils_test.py
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
11 days ago
Update tests/jax_utils_test.py Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
References
#5295 - Remove chex dependency
Author
samanklesaria
Parents
1cdfa4e1
Loading