Closed blacksph3re closed 9 months ago
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.
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