Artem Sergeevich Agarkov

XLand-MiniGrid: Scalable Meta-Reinforcement Learning Environments in JAX

Inspired by the diversity and depth of XLand and the simplicity and minimalism of MiniGrid, we present XLand-MiniGrid, a suite of tools and grid-world environments for meta-reinforcement learning research. Written in JAX, XLand-MiniGrid is designed to be highly scalable and can potentially run on GP…