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