Dependencies in requirements.txt and setup.py are out of sync

[blacksph3re]

Requirements.txt specifies flax==0.7.4, but setup.py requires flax<0.6.2. Unfortunately flax<0.6.2 breaks with jax>1.14, so currently I can't install QDax

[Lookatator]

Hi @blacksph3re, thank you very much for letting us know about this problem!

We have opened a PR which should be merged very soon.

In the meantime you can simply use version v0.2.3 (with git checkout v0.2.3 or pip install qdax==0.2.3) which should be working fine.