search

nissymori / JAX-CORL

Clean single-file implementation of offline RL algorithms in JAX
MIT License
97 stars 2 forks source link

Use jax.tree_util.tree_map #84

Closed nissymori closed 3 months ago

nissymori commented 3 months ago

jax.tree_map cause warning in some jax versions.