Jumanji: a Diverse Suite of Scalable Reinforcement Learning Environments in JAX
Authors:
Clément Bonnet,
Daniel Luo,
Donal Byrne,
Shikha Surana,
Sasha Abramowitz,
Paul Duckworth,
Vincent Coyette,
Laurence I. Midgley,
Elshadai Tegegn,
Tristan Kalloniatis,
Omayma Mahjoub,
Matthew Macfarlane,
Andries P. Smit,
Nathan Grinsztajn,
Raphael Boige,
Cemlyn N. Waters,
Mohamed A. Mimouni,
Ulrich A. Mbou Sob,
Ruan de Kock,
Siddarth Singh,
Daniel Furelos-Blanco,
Victor Le,
Arnu Pretorius,
Alexandre Laterre
Abstract:
Open-source reinforcement learning (RL) environments have played a crucial role in driving progress in the development of AI algorithms. In modern RL research, there is a need for simulated environments that are performant, scalable, and modular to enable their utilization in a wider range of potential real-world applications. Therefore, we present Jumanji, a suite of diverse RL environments speci…
▽ More
Open-source reinforcement learning (RL) environments have played a crucial role in driving progress in the development of AI algorithms. In modern RL research, there is a need for simulated environments that are performant, scalable, and modular to enable their utilization in a wider range of potential real-world applications. Therefore, we present Jumanji, a suite of diverse RL environments specifically designed to be fast, flexible, and scalable. Jumanji provides a suite of environments focusing on combinatorial problems frequently encountered in industry, as well as challenging general decision-making tasks. By leveraging the efficiency of JAX and hardware accelerators like GPUs and TPUs, Jumanji enables rapid iteration of research ideas and large-scale experimentation, ultimately empowering more capable agents. Unlike existing RL environment suites, Jumanji is highly customizable, allowing users to tailor the initial state distribution and problem complexity to their needs. Furthermore, we provide actor-critic baselines for each environment, accompanied by preliminary findings on scaling and generalization scenarios. Jumanji aims to set a new standard for speed, adaptability, and scalability of RL environments.
△ Less
Submitted 15 March, 2024; v1 submitted 16 June, 2023;
originally announced June 2023.
Harmonic functions and the spectrum of the Laplacian on the Sierpinski carpet
Authors:
Matthew Begue,
Tristan Kalloniatis,
Robert S. Strichartz
Abstract:
Kusuoka and Zhou have defined the Laplacian on the Sierpinski carpet using average values of functions on small cells and the graph structure of cell adjacency. We have implemented an algorithm that uses their method to approximate solutions to boundary value problems. As a result we have a wealth of data concerning harmonic functions with prescribed boundary values, and eigenfunctions of the Lapl…
▽ More
Kusuoka and Zhou have defined the Laplacian on the Sierpinski carpet using average values of functions on small cells and the graph structure of cell adjacency. We have implemented an algorithm that uses their method to approximate solutions to boundary value problems. As a result we have a wealth of data concerning harmonic functions with prescribed boundary values, and eigenfunctions of the Laplacian with either Neumann or Dirichlet boundary conditions. We will present some of this data and discuss some ideas for defining normal derivatives on the boundary of the carpet.
△ Less
Submitted 24 January, 2012;
originally announced January 2012.
