LICENSE
README.md
pyproject.toml
setup.py
/Users/puyuan/code/LightZero/lzero/mcts/ctree/ctree_efficientzero/ez_tree.cpp
/Users/puyuan/code/LightZero/lzero/mcts/ctree/ctree_gumbel_muzero/gmz_tree.cpp
/Users/puyuan/code/LightZero/lzero/mcts/ctree/ctree_muzero/mz_tree.cpp
/Users/puyuan/code/LightZero/lzero/mcts/ctree/ctree_muzero_recurrent/mzr_tree.cpp
/Users/puyuan/code/LightZero/lzero/mcts/ctree/ctree_sampled_efficientzero/ezs_tree.cpp
/Users/puyuan/code/LightZero/lzero/mcts/ctree/ctree_stochastic_muzero/stochastic_mz_tree.cpp
/Users/puyuan/code/LightZero/zoo/board_games/gomoku/envs/get_done_winner_cython.cpp
/Users/puyuan/code/LightZero/zoo/board_games/gomoku/envs/legal_actions_cython.cpp
/Users/puyuan/code/LightZero/zoo/board_games/tictactoe/envs/get_done_winner_cython.cpp
/Users/puyuan/code/LightZero/zoo/board_games/tictactoe/envs/legal_actions_cython.cpp
LightZero.egg-info/PKG-INFO
LightZero.egg-info/SOURCES.txt
LightZero.egg-info/dependency_links.txt
LightZero.egg-info/requires.txt
LightZero.egg-info/top_level.txt
lzero/__init__.py
lzero/config/__init__.py
lzero/config/meta.py
lzero/config/utils.py
lzero/entry/__init__.py
lzero/entry/eval_alphazero.py
lzero/entry/eval_muzero.py
lzero/entry/eval_muzero_with_gym_env.py
lzero/entry/train_alphazero.py
lzero/entry/train_muzero.py
lzero/entry/train_muzero_with_gym_env.py
lzero/entry/train_muzero_with_reward_model.py
lzero/entry/utils.py
lzero/envs/__init__.py
lzero/envs/get_wrapped_env.py
lzero/envs/tests/__init__.py
lzero/envs/tests/test_ding_env_wrapper.py
lzero/envs/tests/test_lightzero_env_wrapper.py
lzero/envs/wrappers/__init__.py
lzero/envs/wrappers/action_discretization_env_wrapper.py
lzero/envs/wrappers/lightzero_env_wrapper.py
lzero/mcts/__init__.py
lzero/mcts/utils.py
lzero/mcts/buffer/__init__.py
lzero/mcts/buffer/game_buffer.py
lzero/mcts/buffer/game_buffer_efficientzero.py
lzero/mcts/buffer/game_buffer_gumbel_muzero.py
lzero/mcts/buffer/game_buffer_muzero.py
lzero/mcts/buffer/game_buffer_sampled_efficientzero.py
lzero/mcts/buffer/game_buffer_stochastic_muzero.py
lzero/mcts/buffer/game_segment.py
lzero/mcts/ctree/__init__.py
lzero/mcts/ctree/common_lib/cminimax.cpp
lzero/mcts/ctree/common_lib/cminimax.h
lzero/mcts/ctree/ctree_efficientzero/__init__.py
lzero/mcts/ctree/ctree_efficientzero/lib/cnode.cpp
lzero/mcts/ctree/ctree_efficientzero/lib/cnode.h
lzero/mcts/ctree/ctree_gumbel_muzero/__init__.py
lzero/mcts/ctree/ctree_gumbel_muzero/lib/cnode.cpp
lzero/mcts/ctree/ctree_gumbel_muzero/lib/cnode.h
lzero/mcts/ctree/ctree_muzero/__init__.py
lzero/mcts/ctree/ctree_muzero/lib/cnode.cpp
lzero/mcts/ctree/ctree_muzero/lib/cnode.h
lzero/mcts/ctree/ctree_sampled_efficientzero/__init__.py
lzero/mcts/ctree/ctree_sampled_efficientzero/lib/cnode.cpp
lzero/mcts/ctree/ctree_sampled_efficientzero/lib/cnode.h
lzero/mcts/ctree/ctree_stochastic_muzero/__init__.py
lzero/mcts/ctree/ctree_stochastic_muzero/lib/cnode.cpp
lzero/mcts/ctree/ctree_stochastic_muzero/lib/cnode.h
lzero/mcts/ptree/__init__.py
lzero/mcts/ptree/minimax.py
lzero/mcts/ptree/ptree_az.py
lzero/mcts/ptree/ptree_az_sampled.py
lzero/mcts/ptree/ptree_ez.py
lzero/mcts/ptree/ptree_mz.py
lzero/mcts/ptree/ptree_sez.py
lzero/mcts/ptree/ptree_stochastic_mz.py
lzero/mcts/ptree/test_sez_sample.py
lzero/mcts/tests/__init__.py
lzero/mcts/tests/cprofile_mcts_ptree.py
lzero/mcts/tests/eval_tree_speed.py
lzero/mcts/tests/test_game_buffer.py
lzero/mcts/tests/test_game_segment.py
lzero/mcts/tests/test_image_transform.py
lzero/mcts/tests/test_mcts_ctree.py
lzero/mcts/tests/test_mcts_ptree.py
lzero/mcts/tests/test_mcts_sampled_ctree.py
lzero/mcts/tests/test_utils.py
lzero/mcts/tests/config/__init__.py
lzero/mcts/tests/config/atari_efficientzero_config_for_test.py
lzero/mcts/tests/config/tictactoe_muzero_bot_mode_config_for_test.py
lzero/mcts/tree_search/__init__.py
lzero/mcts/tree_search/mcts_ctree.py
lzero/mcts/tree_search/mcts_ctree_sampled.py
lzero/mcts/tree_search/mcts_ctree_stochastic.py
lzero/mcts/tree_search/mcts_ptree.py
lzero/mcts/tree_search/mcts_ptree_sampled.py
lzero/mcts/tree_search/mcts_ptree_stochastic.py
lzero/model/__init__.py
lzero/model/alphazero_model.py
lzero/model/common.py
lzero/model/efficientzero_model.py
lzero/model/efficientzero_model_mlp.py
lzero/model/image_transform.py
lzero/model/muzero_model.py
lzero/model/muzero_model_mlp.py
lzero/model/sampled_efficientzero_model.py
lzero/model/sampled_efficientzero_model_mlp.py
lzero/model/stochastic_muzero_model.py
lzero/model/stochastic_muzero_model_mlp.py
lzero/model/utils.py
lzero/model/tests/__init__.py
lzero/model/tests/test_alphazero_model.py
lzero/model/tests/test_common.py
lzero/model/tests/test_efficientzero_model.py
lzero/model/tests/test_muzero_model.py
lzero/model/tests/test_sampled_efficientzero_model.py
lzero/model/tests/test_stochastic_muzero_model.py
lzero/policy/__init__.py
lzero/policy/alphazero.py
lzero/policy/efficientzero.py
lzero/policy/gumbel_muzero.py
lzero/policy/muzero.py
lzero/policy/random_policy.py
lzero/policy/sampled_alphazero.py
lzero/policy/sampled_efficientzero.py
lzero/policy/scaling_transform.py
lzero/policy/stochastic_muzero.py
lzero/policy/utils.py
lzero/policy/tests/__init__.py
lzero/policy/tests/test_get_target_obs_index_in_step_k.py
lzero/policy/tests/test_scaling_transform.py
lzero/policy/tests/test_utils.py
lzero/policy/tests/config/__init__.py
lzero/policy/tests/config/atari_muzero_config_for_test.py
lzero/policy/tests/config/cartpole_muzero_config_for_test.py
lzero/worker/__init__.py
lzero/worker/alphazero_collector.py
lzero/worker/alphazero_evaluator.py
lzero/worker/muzero_collector.py
lzero/worker/muzero_evaluator.py
zoo/__init__.py
zoo/atari/__init__.py
zoo/atari/config/__init__.py
zoo/atari/config/atari_efficientzero_config.py
zoo/atari/config/atari_efficientzero_multigpu_ddp_config.py
zoo/atari/config/atari_gumbel_muzero_config.py
zoo/atari/config/atari_muzero_config.py
zoo/atari/config/atari_muzero_multigpu_ddp_config.py
zoo/atari/config/atari_sampled_efficientzero_config.py
zoo/atari/config/atari_stochastic_muzero_config.py
zoo/atari/entry/__init__.py
zoo/atari/entry/atari_eval.py
zoo/atari/envs/__init__.py
zoo/atari/envs/atari_lightzero_env.py
zoo/atari/envs/atari_wrappers.py
zoo/board_games/__init__.py
zoo/board_games/alphabeta_pruning_bot.py
zoo/board_games/mcts_bot.py
zoo/board_games/test_speed_win-rate_between_bots.py
zoo/board_games/classic_search_algorithm/__init__.py
zoo/board_games/classic_search_algorithm/alphabeta_pruning.py
zoo/board_games/classic_search_algorithm/alphabeta_pruning_return_best_subtree.py
zoo/board_games/classic_search_algorithm/minimax_v0.py
zoo/board_games/classic_search_algorithm/minimax_v1.py
zoo/board_games/go/__init__.py
zoo/board_games/go/config/__init__.py
zoo/board_games/go/config/go_alphazero_sp-mode_config.py
zoo/board_games/go/envs/__init__.py
zoo/board_games/go/envs/go_env.py
zoo/board_games/go/envs/test_go_env.py
zoo/board_games/gomoku/__init__.py
zoo/board_games/gomoku/config/__init__.py
zoo/board_games/gomoku/config/gomoku_alphazero_bot_mode_config.py
zoo/board_games/gomoku/config/gomoku_alphazero_sp_mode_config.py
zoo/board_games/gomoku/config/gomoku_gumbel_muzero_bot_mode_config.py
zoo/board_games/gomoku/config/gomoku_muzero_bot_mode_config.py
zoo/board_games/gomoku/config/gomoku_muzero_sp_mode_config.py
zoo/board_games/gomoku/config/gomoku_sampled_alphazero_bot_mode_config.py
zoo/board_games/gomoku/config/gomoku_sampled_alphazero_sp_mode_config.py
zoo/board_games/gomoku/entry/__init__.py
zoo/board_games/gomoku/entry/gomoku_alphazero_eval.py
zoo/board_games/gomoku/entry/gomoku_gumbel_muzero_eval.py
zoo/board_games/gomoku/entry/gomoku_muzero_eval.py
zoo/board_games/gomoku/entry/gomoku_sampled_alphazero_eval.py
zoo/board_games/gomoku/envs/__init__.py
zoo/board_games/gomoku/envs/gomoku_env.py
zoo/board_games/gomoku/envs/gomoku_human_vs_bot_UI.py
zoo/board_games/gomoku/envs/gomoku_rule_bot_v0.py
zoo/board_games/gomoku/envs/gomoku_rule_bot_v1.py
zoo/board_games/gomoku/envs/test_gomoku_alphabeta_pruning_bot.py
zoo/board_games/gomoku/envs/test_gomoku_env.py
zoo/board_games/gomoku/envs/test_gomoku_mcts_bot.py
zoo/board_games/gomoku/envs/test_gomoku_rule_bot_v0.py
zoo/board_games/gomoku/envs/test_gomoku_rule_bot_v1.py
zoo/board_games/gomoku/envs/utils.py
zoo/board_games/gomoku/test/__init__.py
zoo/board_games/gomoku/test/eval_get_done_winner.py
zoo/board_games/gomoku/test/eval_legal_actions.py
zoo/board_games/gomoku/test/test_get_done_winner_cython.py
zoo/board_games/gomoku/test/test_gomoku_env_legal_actions.py
zoo/board_games/gomoku/test/test_legal_actions_cython.py
zoo/board_games/tictactoe/__init__.py
zoo/board_games/tictactoe/config/__init__.py
zoo/board_games/tictactoe/config/tictactoe_alphazero_bot_mode_config.py
zoo/board_games/tictactoe/config/tictactoe_alphazero_bot_mode_multigpu_ddp_config.py
zoo/board_games/tictactoe/config/tictactoe_alphazero_sp_mode_config.py
zoo/board_games/tictactoe/config/tictactoe_alphazero_sp_mode_multigpu_ddp_config.py
zoo/board_games/tictactoe/config/tictactoe_gumbel_muzero_bot_mode_config.py
zoo/board_games/tictactoe/config/tictactoe_muzero_bot_mode_config.py
zoo/board_games/tictactoe/config/tictactoe_muzero_sp_mode_config.py
zoo/board_games/tictactoe/entry/__init__.py
zoo/board_games/tictactoe/entry/tictactoe_alphazero_eval.py
zoo/board_games/tictactoe/entry/tictactoe_muzero_eval.py
zoo/board_games/tictactoe/envs/__init__.py
zoo/board_games/tictactoe/envs/test_tictactoe_alphabeta_prunning_bot.py
zoo/board_games/tictactoe/envs/test_tictactoe_env.py
zoo/board_games/tictactoe/envs/test_tictactoe_mcts_bot.py
zoo/board_games/tictactoe/envs/test_tictactoe_rule_bot_v0.py
zoo/board_games/tictactoe/envs/tictactoe_env.py
zoo/box2d/__init__.py
zoo/box2d/bipedalwalker/__init__.py
zoo/box2d/bipedalwalker/config/__init__.py
zoo/box2d/bipedalwalker/config/bipedalwalker_cont_disc_efficientzero_config.py
zoo/box2d/bipedalwalker/config/bipedalwalker_cont_disc_sampled_efficientzero_config.py
zoo/box2d/bipedalwalker/config/bipedalwalker_cont_sampled_efficientzero_config.py
zoo/box2d/bipedalwalker/config/tmp.py
zoo/box2d/bipedalwalker/entry/__init__.py
zoo/box2d/bipedalwalker/entry/bipedalwalker_eval.py
zoo/box2d/bipedalwalker/envs/__init__.py
zoo/box2d/bipedalwalker/envs/bipedalwalker_cont_disc_env.py
zoo/box2d/bipedalwalker/envs/bipedalwalker_env.py
zoo/box2d/lunarlander/__init__.py
zoo/box2d/lunarlander/config/__init__.py
zoo/box2d/lunarlander/config/lunarlander_cont_disc_efficientzero_config.py
zoo/box2d/lunarlander/config/lunarlander_cont_disc_sampled_efficientzero_config.py
zoo/box2d/lunarlander/config/lunarlander_cont_sampled_efficientzero_config.py
zoo/box2d/lunarlander/config/lunarlander_disc_efficientzero_config.py
zoo/box2d/lunarlander/config/lunarlander_disc_gumbel_muzero_config.py
zoo/box2d/lunarlander/config/lunarlander_disc_muzero_config.py
zoo/box2d/lunarlander/config/lunarlander_disc_stochastic_muzero_config.py
zoo/box2d/lunarlander/entry/__init__.py
zoo/box2d/lunarlander/entry/lunarlander_eval.py
zoo/box2d/lunarlander/envs/__init__.py
zoo/box2d/lunarlander/envs/lunarlander_cont_disc_env.py
zoo/box2d/lunarlander/envs/lunarlander_env.py
zoo/box2d/lunarlander/envs/test_lunarlander_env.py
zoo/classic_control/__init__.py
zoo/classic_control/cartpole/__init__.py
zoo/classic_control/cartpole/config/__init__.py
zoo/classic_control/cartpole/config/cartpole_efficientzero_config.py
zoo/classic_control/cartpole/config/cartpole_gumbel_muzero_config.py
zoo/classic_control/cartpole/config/cartpole_muzero_config.py
zoo/classic_control/cartpole/config/cartpole_muzero_config_ckpt.py
zoo/classic_control/cartpole/config/cartpole_sampled_efficientzero_config.py
zoo/classic_control/cartpole/config/cartpole_stochastic_muzero_config.py
zoo/classic_control/cartpole/entry/__init__.py
zoo/classic_control/cartpole/entry/cartpole_eval.py
zoo/classic_control/cartpole/envs/__init__.py
zoo/classic_control/cartpole/envs/cartpole_lightzero_env.py
zoo/classic_control/pendulum/__init__.py
zoo/classic_control/pendulum/config/__init__.py
zoo/classic_control/pendulum/config/pendulum_cont_disc_efficientzero_config.py
zoo/classic_control/pendulum/config/pendulum_cont_disc_gumbel_muzero_config.py
zoo/classic_control/pendulum/config/pendulum_cont_disc_muzero_config.py
zoo/classic_control/pendulum/config/pendulum_cont_disc_sampled_efficientzero_config.py
zoo/classic_control/pendulum/config/pendulum_cont_sampled_efficientzero_config.py
zoo/classic_control/pendulum/entry/__init__.py
zoo/classic_control/pendulum/entry/pendulum_eval.py
zoo/classic_control/pendulum/envs/__init__.py
zoo/classic_control/pendulum/envs/pendulum_lightzero_env.py
zoo/game_2048/__init__.py
zoo/game_2048/entry/2048_bot_eval.py
zoo/game_2048/entry/2048_eval.py
zoo/game_2048/entry/__init__.py
zoo/game_2048/envs/__init__.py
zoo/game_2048/envs/expectimax_search_based_bot.py
zoo/game_2048/envs/game_2048_env.py
zoo/game_2048/envs/test_game_2048_env.py
zoo/minigrid/__init__.py
zoo/minigrid/config/__init__.py
zoo/minigrid/config/minigrd_sampled_efficientzero_config.py
zoo/minigrid/config/minigrid_efficientzero_config.py
zoo/minigrid/config/minigrid_muzero_config.py
zoo/minigrid/config/minigrid_muzero_rnd_config.py
zoo/minigrid/entry/__init__.py
zoo/minigrid/entry/minigrid_eval.py