search

FLAIROx / JaxMARL

Multi-Agent Reinforcement Learning with JAX
Apache License 2.0
418 stars 76 forks source link

Jax versioning #116

Open sash-a opened 2 days ago

sash-a commented 2 days ago

Hey guys :wave:

Just wondering if there's any reason you pin Jax to 0.4.25? We are having issues on Mava using JaxMARL as it uses a newer version of Jax and soon will be incompatible with 0.4.25. I don't think it would cause any issues if you unpin, but I haven't done a thorough check.

The culprit PR on mava: https://github.com/instadeepai/Mava/pull/1103

wangreit commented 11 hours ago

Hi!

I am getting the following errors when running the colab walkthrough. I think it's also related to the incompatible issues with Jax, Jaxlib, and flax.

ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts. accelerate 0.34.2 requires safetensors>=0.4.3, but you have safetensors 0.4.2 which is incompatible. dopamine-rl 4.0.9 requires gym<=0.25.2, but you have gym 0.26.2 which is incompatible.

Could you please take a look?

Thanks!