No module named 'jax.experimental.gda_serialization'

Open Amazing-Ming opened this issue 2 years ago • 4 comments

Jun 05 '23 03:06 Amazing-Ming

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

Jun 14 '23 09:06 Chunyuan-Li

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

Jun 17 '23 13:06 lvxiangyi

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

Jun 17 '23 20:06 tikhomirovd

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' {} +

Jun 23 '23 10:06 shuntacurosu