NAVIX: Scaling MiniGrid Environments with JAX - Pignatelli, Eduardo; Liesen, Jarek; Lange, Robert; Lu, Chris; Castro, Pablo Samuel; Toni, Laura (Advances in Neural Information Processing Systems, 2025)