Closed Alnusjaponica closed 8 months ago
Remove version specification on haiku and resolve CI failure
This PR removes version specification of jax and jaxlib from requirements.txt Note that the reason to pin the version, jax_experimental_name_stack option, has been resolved at Haiku v0.0.9 For more details see https://github.com/optuna/optuna-examples/pull/144 and https://github.com/google-deepmind/dm-haiku/releases/tag/v0.0.9.
jax
jaxlib
requirements.txt
jax_experimental_name_stack
Haiku v0.0.9
Motivation
Remove version specification on haiku and resolve CI failure
Description of the changes
This PR removes version specification of
jaxandjaxlibfromrequirements.txtNote that the reason to pin the version,jax_experimental_name_stackoption, has been resolved atHaiku v0.0.9For more details see https://github.com/optuna/optuna-examples/pull/144 and https://github.com/google-deepmind/dm-haiku/releases/tag/v0.0.9.