flax pin versions (#24791)

Commit

2 years ago

Upgrade jax/jaxlib/flax pin versions (#24791) fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

References

Author

ydshieh

Parents