Maybe specify specific package version in setup.py?

[Roadelse]

I am trying to run the demo, but failed with: ImportError: cannot import name 'config' from 'jax.config' (.../envs/jax-am/lib/python3.9/site-packages/jax/config.py)

It seems to be an error from mismatched (i.e., newer) jax version, and I find the required packages in setup.py just decalre the package name without detailed version.

So it would be better to specify the package version in setup.py?