Would it be possible for you to provide the specific version numbers of the following libraries: jax, haiku, optax, scipy? 