Bump jax and flax version upper bounds (#572)
#1859
| Job | Run time |
|---|---|
| 57s | |
| 57s |
jax and flax version upper bounds (#572)
#1859
| Job | Run time |
|---|---|
| 57s | |
| 57s |