'Config' object has no attribute 'jax_experimental_name_stack'
VGPReys opened this issue · comments
I do have the same issue as the one in #3.
By running the example python main.py, I get the 'Config' object has no attribute 'jax_experimental_name_stack' error from jax.
Python 3.8.17 (default, Jul 5 2023, 21:04:15)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
import jax
jax.__version__'0.4.13'
How should I solve this ?
Hi, can you update haiku to 0.0.10 and have another try?
Hi,
Indeed, it solved my problem !
Thanks a lot.