instadeepai/jumanji

Features

• Algorithm Implementations - Hardware-accelerated environments for industry-driven research.
• Reinforcement Learning - Industry-driven hardware-accelerated reinforcement learning environments.
• Reinforcement Learning Frameworks - Hardware-accelerated reinforcement learning environments built with JAX.