Available Environments¤
Jaxnasium doesn't bundle a large number of environments directly. Instead, it relies on existing work and wraps these environments in various wrappers to conform to the Jaxnasium API. This approach allows users to leverage a wide array of established environments while maintaining the performance and flexibility offered by JAX.
Native Environments¤
Note: The classic control environments are natively implemented ánd bundled in Jaxnasium convenience:
CartPole-v1MountainCar-v0Acrobot-v1Pendulum-v1MountainCarContinuous-v0
These environments are implemented directly in Jaxnasium and don't require external dependencies.
External Environments using the Jaxnasium API¤
These environments run without wrappers.
External Environment Libraries¤
Jaxnasium integrates with the following external environment libraries through wrapper adapters. See the end of this page for a full list of available environments.
Gymnax¤
JAX implementations of OpenAI's Gym environments, offering accelerated and parallelized rollouts. Includes classic control, bsuite, and MinAtar environments.
Jumanji¤
A suite of diverse, scalable reinforcement learning environments implemented in JAX by DeepMind. Focuses on combinatorial problems and general decision-making tasks.
Brax¤
A fast and flexible physics simulation engine for training and evaluating rigid body environments in JAX by Google.
Pgx¤
JAX implementations of various board games and classic environments, including chess, Go, shogi, and more.
JaxMARL¤
Multi-agent reinforcement learning environments implemented in JAX, including MPE (Multi-Particle Environment) scenarios and other multi-agent tasks.
xMinigrid¤
JAX implementation of MiniGrid environments, including XLand variants for procedural generation research.
Navix¤
JAX implementation of navigation environments, providing various gridworld navigation tasks.
Craftax¤
JAX implementation of Craftax environments, inspired by Minecraft-like crafting and survival tasks.
Usage¤
To use any of these environments, simply call:
import jaxnasium as jym
# Native environments
env = jym.make("CartPole-v1")
# External environments (requires installing the respective library)
env = jym.make("Breakout-MinAtar") # Gymnax
env = jym.make("Game2048-v1") # Jumanji
env = jym.make("ant") # Brax
env = jym.make("chess") # Pgx
Note: External environment libraries are not bundled as dependencies and need to be installed manually (e.g., via pip) before use.
Complete List of Registered Environments¤
Below is the complete list of all registered environment strings available in Jaxnasium:
Auto-generated List
This list is automatically generated from the Jaxnasium registry.
Acrobot-v1CartPole-v1MountainCarContinuous-v0MountainCar-v0Pendulum-v1gymnax:CartPole-v1gymnax:Acrobot-v1gymnax:Pendulum-v1gymnax:MountainCar-v0gymnax:MountainCarContinuous-v0gymnax:Asterix-MinAtargymnax:Breakout-MinAtargymnax:Freeway-MinAtargymnax:SpaceInvaders-MinAtargymnax:DeepSea-bsuitegymnax:Catch-bsuitegymnax:MemoryChain-bsuitegymnax:UmbrellaChain-bsuitegymnax:DiscountingChain-bsuitegymnax:MNISTBandit-bsuitegymnax:SimpleBandit-bsuitegymnax:FourRooms-miscgymnax:MetaMaze-miscgymnax:PointRobot-miscgymnax:BernoulliBandit-miscgymnax:GaussianBandit-miscgymnax:Reacher-miscgymnax:Swimmer-miscgymnax:Pong-miscjumanji:Game2048-v1jumanji:GraphColoring-v1jumanji:Minesweeper-v0jumanji:RubiksCube-v0jumanji:RubiksCube-partly-scrambled-v0jumanji:SlidingTilePuzzle-v0jumanji:Sudoku-v0jumanji:Sudoku-very-easy-v0jumanji:BinPack-v2jumanji:FlatPack-v0jumanji:JobShop-v0jumanji:Knapsack-v1jumanji:Tetris-v0jumanji:Cleaner-v0jumanji:Connector-v2jumanji:CVRP-v1jumanji:MultiCVRP-v0jumanji:Maze-v0jumanji:RobotWarehouse-v0jumanji:Snake-v1jumanji:TSP-v1jumanji:MMST-v0jumanji:PacMan-v1jumanji:Sokoban-v0jumanji:LevelBasedForaging-v0jumanji:SearchAndRescue-v0brax:antbrax:halfcheetahbrax:hopperbrax:humanoidbrax:humanoidstandupbrax:inverted_pendulumbrax:inverted_double_pendulumbrax:pusherbrax:reacherbrax:walker2dpgx:2048pgx:animal_shogipgx:backgammonpgx:chesspgx:connect_fourpgx:gardner_chesspgx:go_9x9pgx:go_19x19pgx:hexpgx:kuhn_pokerpgx:leduc_holdempgx:minatar-asterixpgx:minatar-breakoutpgx:minatar-freewaypgx:minatar-seaquestpgx:minatar-space_invaderspgx:othellopgx:shogipgx:sparrow_mahjongpgx:tic_tac_toejaxmarl:MPE_simple_v3jaxmarl:MPE_simple_tag_v3jaxmarl:MPE_simple_world_comm_v3jaxmarl:MPE_simple_spread_v3jaxmarl:MPE_simple_crypto_v3jaxmarl:MPE_simple_speaker_listener_v4jaxmarl:MPE_simple_push_v3jaxmarl:MPE_simple_adversary_v3jaxmarl:MPE_simple_reference_v3jaxmarl:MPE_simple_facmac_v1jaxmarl:MPE_simple_facmac_3a_v1jaxmarl:MPE_simple_facmac_6a_v1jaxmarl:MPE_simple_facmac_9a_v1jaxmarl:switch_riddlejaxmarl:SMAXjaxmarl:HeuristicEnemySMAXjaxmarl:ant_4x2jaxmarl:halfcheetah_6x1jaxmarl:hopper_3x1jaxmarl:humanoid_9|8jaxmarl:walker2d_2x3jaxmarl:stormjaxmarl:storm_2pjaxmarl:storm_npjaxmarl:hanabijaxmarl:overcookedjaxmarl:overcooked_v2jaxmarl:coin_gamejaxmarl:jaxnavxminigrid:XLand-MiniGrid-R1-9x9xminigrid:XLand-MiniGrid-R1-11x11xminigrid:XLand-MiniGrid-R1-13x13xminigrid:XLand-MiniGrid-R1-15x15xminigrid:XLand-MiniGrid-R1-17x17xminigrid:XLand-MiniGrid-R2-9x9xminigrid:XLand-MiniGrid-R2-11x11xminigrid:XLand-MiniGrid-R2-13x13xminigrid:XLand-MiniGrid-R2-15x15xminigrid:XLand-MiniGrid-R2-17x17xminigrid:XLand-MiniGrid-R4-9x9xminigrid:XLand-MiniGrid-R4-11x11xminigrid:XLand-MiniGrid-R4-13x13xminigrid:XLand-MiniGrid-R4-15x15xminigrid:XLand-MiniGrid-R4-17x17xminigrid:XLand-MiniGrid-R6-13x13xminigrid:XLand-MiniGrid-R6-17x17xminigrid:XLand-MiniGrid-R6-19x19xminigrid:XLand-MiniGrid-R9-16x16xminigrid:XLand-MiniGrid-R9-19x19xminigrid:XLand-MiniGrid-R9-25x25xminigrid:MiniGrid-BlockedUnlockPickUpxminigrid:MiniGrid-DoorKey-5x5xminigrid:MiniGrid-DoorKey-6x6xminigrid:MiniGrid-DoorKey-8x8xminigrid:MiniGrid-DoorKey-16x16xminigrid:MiniGrid-Empty-5x5xminigrid:MiniGrid-Empty-6x6xminigrid:MiniGrid-Empty-8x8xminigrid:MiniGrid-Empty-16x16xminigrid:MiniGrid-EmptyRandom-5x5xminigrid:MiniGrid-EmptyRandom-6x6xminigrid:MiniGrid-EmptyRandom-8x8xminigrid:MiniGrid-EmptyRandom-16x16xminigrid:MiniGrid-FourRoomsxminigrid:MiniGrid-LockedRoomxminigrid:MiniGrid-MemoryS8xminigrid:MiniGrid-MemoryS16xminigrid:MiniGrid-MemoryS32xminigrid:MiniGrid-MemoryS64xminigrid:MiniGrid-MemoryS128xminigrid:MiniGrid-Playgroundxminigrid:MiniGrid-Unlockxminigrid:MiniGrid-UnlockPickUpnavix:Navix-Empty-5x5-v0navix:Navix-Empty-6x6-v0navix:Navix-Empty-8x8-v0navix:Navix-Empty-16x16-v0navix:Navix-Empty-Random-5x5-v0navix:Navix-Empty-Random-6x6-v0navix:Navix-Empty-Random-8x8-v0navix:Navix-Empty-Random-16x16-v0navix:Navix-DoorKey-5x5-v0navix:Navix-DoorKey-6x6-v0navix:Navix-DoorKey-8x8-v0navix:Navix-DoorKey-16x16-v0navix:Navix-DoorKey-Random-5x5-v0navix:Navix-DoorKey-Random-6x6-v0navix:Navix-DoorKey-Random-8x8-v0navix:Navix-DoorKey-Random-16x16-v0navix:Navix-FourRooms-v0navix:Navix-KeyCorridorS3R1-v0navix:Navix-KeyCorridorS3R2-v0navix:Navix-KeyCorridorS3R3-v0navix:Navix-KeyCorridorS4R3-v0navix:Navix-KeyCorridorS5R3-v0navix:Navix-KeyCorridorS6R3-v0navix:Navix-LavaGapS5-v0navix:Navix-LavaGapS6-v0navix:Navix-LavaGapS7-v0navix:Navix-SimpleCrossingS9N1-v0navix:Navix-SimpleCrossingS9N2-v0navix:Navix-SimpleCrossingS9N3-v0navix:Navix-SimpleCrossingS11N5-v0navix:Navix-Dynamic-Obstacles-5x5-v0navix:Navix-Dynamic-Obstacles-5x5-Random-v0navix:Navix-Dynamic-Obstacles-6x6-v0navix:Navix-Dynamic-Obstacles-6x6-Random-v0navix:Navix-Dynamic-Obstacles-8x8-v0navix:Navix-Dynamic-Obstacles-16x16-v0navix:Navix-DistShift1-v0navix:Navix-DistShift2-v0navix:Navix-GoToDoor-5x5-v0navix:Navix-GoToDoor-6x6-v0navix:Navix-GoToDoor-8x8-v0craftax:Craftax-Classic-Symbolic-v1craftax:Craftax-Classic-Pixels-v1craftax:Craftax-Symbolic-v1craftax:Craftax-Pixels-v1