Skip to content

Restrict JAX version to 0.3.6

Vicentini Filippo requested to merge jax-version-0.3.6 into master

Created by: femtobit

jax 0.3.7 is incompatible with the currently released chex which optax depends on (due to deepmind/chex#156).

Merge request reports