Fix `jax` version to 0.4.28

tensorflow / datasets

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

https://www.tensorflow.org/datasets

Apache License 2.0

4.23k stars 1.52k forks source link

Closed copybara-service[bot] closed 3 weeks ago

copybara-service[bot] commented 3 weeks ago

Fix jax version to 0.4.28