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 GPU or TPU accelerators, democratizing large-scale experimentation with limited resources. Along with the environments, XLand-MiniGrid provides pre-sampled benchmarks with millions of unique tasks of varying difficulty and easy-to-use baselines that allow users to quickly start training adaptive agents. In addition, we have conducted a preliminary analysis of scaling and generalization, showing that our baselines are capable of reaching millions of steps per second during training and validating that the proposed benchmarks are challenging. XLand-MiniGrid is open-source and available at https://github.com/dunnolab/xland-minigrid.
함께 살펴볼 논문
공통 태그를 근거로 연결했습니다. 실제 인용 관계나 논문의 우수성을 뜻하지 않습니다.
Meta Reinforcement Learning for Multi-Task Offloading in Vehicular Edge Computing
Step-wise Policy for Rare-tool Knowledge (SPaRK): Offline RL that Drives Diverse Tool Use in LLMs
AutoTriton: Automatic Triton Programming with Reinforcement Learning in LLMs
DIP: Unsupervised Dense In-Context Post-training of Visual Representations
CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learning
Mixture of Experts in Large Language Models
토론
이 논문과 연결된 포럼 글이 아직 없습니다.