Can you share the versions of libraries

[gyq13]

I'm interested in your project, but it's hard to run the code without specific library version.Can you share the versions of libraries like numpy, jax, jaxlib,optax,haiku,etc. Many thanks.

[peterparity]

This should be the list of relevant library versions that I used:

dm-haiku==0.0.4.dev0 jax==0.2.8 jaxlib==0.1.59+cuda110 numpy==1.19.2 optax==0.0.2 tqdm==4.55.1

I also attached my full environment here: requirements.txt

Let me know if you run into any issues! (If it's breaking with later versions, let me know and I can try to fix it.)

[fangkaiqi]

This should be the list of relevant library versions that I used:

dm-haiku==0.0.4.dev0 jax==0.2.8 jaxlib==0.1.59+cuda110 numpy==1.19.2 optax==0.0.2 tqdm==4.55.1

I also attached my full environment here: requirements.txt

Let me know if you run into any issues! (If it's breaking with later versions, let me know and I can try to fix it.) Hello, Doctor. I've recently been learning to reproduce this article, probably due to the version, which has led me to two problems, which I mentioned in my posted question, do you have time to help with this?