No module named 'jax.experimental.gda_serialization'

[Amazing-Ming]

No module named 'jax.experimental.gda_serialization'

[Chunyuan-Li]

update jax and t5x，(may need to uninstall package [jestimator] first)

[lvxiangyi]

I also encountered this problem，能不能说的更详细一点

[tikhomirovd]

It's renamed in new version jax to jax.experimental.array_serialization

[shuntacurosu]

This issue can be temporarily resolved by renaming the package name.
To do this, run the following block immediately before the "Imports and Definitions" section.

#@title Replace package name for t5x
!find /usr/local/lib/python3.10/dist-packages/orbax -type f -name "*.py" -exec sed -i 's/gda_serialization/array_serialization/g' {} +