corl-team / xland-minigrid

JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️
Apache License 2.0
195 stars 15 forks source link

RGB image observation wrapper compatible with jit #9

Closed Howuhh closed 7 months ago

Howuhh commented 8 months ago

see https://github.com/corl-team/xland-minigrid/issues/8

Howuhh commented 8 months ago

Will regenerate all benchmarks for this...

vkurenkov commented 8 months ago

I think we should add documentation on this to readme before merging