search

instadeepai / jumanji

🕹️ A diverse suite of scalable reinforcement learning environments in JAX
https://instadeepai.github.io/jumanji
Apache License 2.0
584 stars 71 forks source link

fix: latest chex and jax #181

Closed clement-bonnet closed 1 year ago

clement-bonnet commented 1 year ago

With chex>0.1.5, chex.Array is not compatible with isinstance used in training.