23 September error : AttributeError: 'Config' object has no attribute 'jax_experimental_name_stack'

[f3mf3m]

Hi

tried mono or mulit, same issue

Thank's

FH

[kidming]

Yeah, same problem.

[anupam-tiwari]

This is an upgrade from google, the newest version of colab by default takes up jax 0.3.25 (https://colab.research.google.com/notebooks/relnotes.ipynb#scrollTo=iOWrh41ocOca) which has no support for 'jax_experimental_name_stack', I have solved this issue by downgrading colab jax to 0.3.17, you can do it by adding this command: !pip install --upgrade "jax[cuda]==0.3.17" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html and then restarting the runtime

[kidming]

I tried it, jax goes back to 0.3.17 but it comes with other problems. May I ask which line you put in this commands?