Jumanji: a Diverse Suite of Scalable Reinforcement Learning Environments in JAX.

Clément Bonnet Daniel Luo Donal Byrne Shikha Surana Sasha Abramowitz Paul Duckworth Vincent Coyette Laurence Illing Midgley Elshadai Tegegn Tristan Kalloniatis Omayma Mahjoub Matthew Macfarlane Andries P. Smit Nathan Grinsztajn Raphaël 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

Published in: ICLR (2024)

Keyphrases

learning environment
reinforcement learning
learning activities
wide variety
learning systems
highly scalable
learning process
learning processes
databases
real world
pedagogical model
data sets
rich set
scalable video coding
adaptive systems
memory efficient
learning outcomes
lightweight
online learning
e learning