corl-team / xland-minigrid

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

RGB image observation wrapper compatible with jit #9

Closed Howuhh closed 3 months ago

Howuhh commented 4 months ago

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

Howuhh commented 4 months ago

Will regenerate all benchmarks for this...

vkurenkov commented 4 months ago

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