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

19 December 2023

Papers citing "XLand-MiniGrid: Scalable Meta-Reinforcement Learning Environments in JAX"

26 / 26 papers shown

Title
Zero-Shot Adaptation of Behavioral Foundation Models to Unseen Dynamics Maksim Bobrin Ilya Zisman Alexander Nikulin Vladislav Kurenkov Dmitry V. Dylov OffRL 17 0 0 19 May 2025
Meta-World+: An Improved, Standardized, RL Benchmark Reginald McLean Evangelos Chatzaroulas Luc McCutcheon Frank Röder Tianhe Yu ... Ryan Julian Jordan Terry Isaac Woungang Nariman Farsad Pablo Samuel Castro OffRL 21 0 0 16 May 2025
Cross-environment Cooperation Enables Zero-shot Multi-agent Coordination Kunal Jha Wilka Carvalho Yancheng Liang S. Du Max Kleiman-Weiner Natasha Jaques 35 0 0 17 Apr 2025
Thinking agents for zero-shot generalization to qualitatively novel tasks Thomas Miconi Kevin L McKee Yicong Zheng Jed McCaleb LRM AI4CE 53 0 0 25 Mar 2025
Partially Observable Reinforcement Learning with Memory Traces Onno Eberhard Michael Muehlebach Claire Vernade OffRL 66 0 0 19 Mar 2025
SocialJax: An Evaluation Suite for Multi-agent Reinforcement Learning in Sequential Social Dilemmas Zihao Guo Richard Willis Richard Willis Tristan Tomilin Joel Z Leibo Yali Du 63 0 0 18 Mar 2025
Yes, Q-learning Helps Offline In-Context RL Denis Tarasov Alexander Nikulin Ilya Zisman Albina Klepach Andrei Polubarov Nikita Lyubaykin Alexander Derevyagin Igor Kiselev Vladislav Kurenkov OffRL OnRL 255 1 0 24 Feb 2025
A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks Thomas Schmied Thomas Adler Vihang Patil M. Beck Korbinian Poppel Johannes Brandstetter Günter Klambauer Razvan Pascanu Sepp Hochreiter 79 5 0 21 Feb 2025
AMAGO-2: Breaking the Multi-Task Barrier in Meta-Reinforcement Learning with Transformers Jake Grigsby Justin Sasek Samyak Parajuli Daniel Adebi Amy Zhang Yuke Zhu OffRL 30 3 0 17 Nov 2024
N-Gram Induction Heads for In-Context RL: Improving Stability and Reducing Data Needs Ilya Zisman Alexander Nikulin Andrei Polubarov Nikita Lyubaykin Vladislav Kurenkov Andrei Polubarov Igor Kiselev Vladislav Kurenkov OffRL 61 2 0 04 Nov 2024
Kinetix: Investigating the Training of General Agents through Open-Ended Physics-Based Control Tasks Michael T. Matthews Michael Beukman Chris Xiaoxuan Lu Jakob Foerster OffRL AI4CE 41 3 0 30 Oct 2024
No Regrets: Investigating and Improving Regret Approximations for Curriculum Discovery Alexander Rutherford Michael Beukman Timon Willi Bruno Lacerda Nick Hawes Jakob Foerster 53 6 0 27 Aug 2024
NAVIX: Scaling MiniGrid Environments with JAX Eduardo Pignatelli Jarek Liesen R. T. Lange Chris Xiaoxuan Lu Pablo Samuel Castro Laura Toni 47 3 0 28 Jul 2024
RobocupGym: A challenging continuous control benchmark in Robocup Michael Beukman Branden Ingram Geraud Nangue Tasse Benjamin Rosman Pravesh Ranchod OffRL 45 1 0 03 Jul 2024
The Overcooked Generalisation Challenge Constantin Ruhdorfer Matteo Bortoletto Anna Penzkofer Andreas Bulling 51 4 0 25 Jun 2024
XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning Alexander Nikulin Ilya Zisman Alexey Zemtsov Viacheslav Sinii 120 5 0 13 Jun 2024
Massively Multiagent Minigames for Training Generalist Agents Kyoung Whan Choe Ryan Sullivan Joseph Suárez AI4CE 36 0 0 07 Jun 2024
A Batch Sequential Halving Algorithm without Performance Degradation Sotetsu Koyamada Soichiro Nishimori Shin Ishii 32 0 0 01 Jun 2024
Benchmarking General-Purpose In-Context Learning Fan Wang Chuan Lin Yang Cao Yu Kang 43 1 0 27 May 2024
Craftax: A Lightning-Fast Benchmark for Open-Ended Reinforcement Learning Michael T. Matthews Michael Beukman Benjamin Ellis Mikayel Samvelyan Matthew Jackson Samuel Coward Jakob Foerster OffRL 42 26 0 26 Feb 2024
In-Context Reinforcement Learning for Variable Action Spaces Viacheslav Sinii Alexander Nikulin Vladislav Kurenkov Ilya Zisman Sergey Kolesnikov 29 14 0 20 Dec 2023
Emergence of In-Context Reinforcement Learning from Noise Distillation Ilya Zisman Vladislav Kurenkov Alexander Nikulin Viacheslav Sinii Sergey Kolesnikov OffRL 43 10 0 19 Dec 2023
minimax: Efficient Baselines for Autocurricula in JAX Minqi Jiang Michael Dennis Edward Grefenstette Tim Rocktaschel 32 8 0 21 Nov 2023
Structured State Space Models for In-Context Reinforcement Learning Chris Xiaoxuan Lu Yannick Schroecker Albert Gu Emilio Parisotto Jakob N. Foerster Satinder Singh Feryal M. P. Behbahani AI4TS 102 84 0 07 Mar 2023
Grounding Language to Entities and Dynamics for Generalization in Reinforcement Learning H. Wang Victor Zhong Karthik Narasimhan 81 54 0 19 Jan 2021
Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks Chelsea Finn Pieter Abbeel Sergey Levine OOD 496 11,727 0 09 Mar 2017