**ImportError: cannot import name ‘optim’ from ‘flax’ 原因: flax 0.6.0把optim模块移除了解决方法:安装老版本就可以啦可以看看:https://github.com/google/flax/commit/cda7a4c85bbce744e412ab82e298ddf76d4770d2