├── .gitignore ├── LICENSE ├── README.md ├── agents ├── BaseAgent.py ├── DDPG.py ├── DDQN.py ├── DQN.py ├── MaxminDQN.py ├── NAF.py ├── PPO.py ├── SAC.py ├── TD3.py └── __init__.py ├── analysis.py ├── clean.sh ├── components ├── networks.py └── replay.py ├── configs ├── classic_ddqn.json ├── classic_dqn.json ├── classic_maxmin.json ├── lunar_ddqn.json ├── lunar_dqn.json ├── lunar_maxmin.json ├── minatar_ddqn.json ├── minatar_dqn.json ├── minatar_maxmin.json ├── mujoco_ddpg.json ├── mujoco_naf.json ├── mujoco_ppo.json ├── mujoco_sac.json ├── mujoco_td3.json ├── nchain_dqn.json ├── pygame_ddqn.json ├── pygame_dqn.json └── pygame_maxmin.json ├── envs ├── env.py └── wrappers.py ├── experiment.py ├── find_config.py ├── git_commit_id.sh ├── main.py ├── move_log.sh ├── requirements.txt ├── results └── performance.md ├── sbatch_m.sh ├── sbatch_s.sh ├── submit.py └── utils ├── helper.py ├── logger.py ├── plotter.py ├── submitter.py └── sweeper.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/README.md -------------------------------------------------------------------------------- /agents/BaseAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/agents/BaseAgent.py -------------------------------------------------------------------------------- /agents/DDPG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/agents/DDPG.py -------------------------------------------------------------------------------- /agents/DDQN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/agents/DDQN.py -------------------------------------------------------------------------------- /agents/DQN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/agents/DQN.py -------------------------------------------------------------------------------- /agents/MaxminDQN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/agents/MaxminDQN.py -------------------------------------------------------------------------------- /agents/NAF.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/agents/NAF.py -------------------------------------------------------------------------------- /agents/PPO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/agents/PPO.py -------------------------------------------------------------------------------- /agents/SAC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/agents/SAC.py -------------------------------------------------------------------------------- /agents/TD3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/agents/TD3.py -------------------------------------------------------------------------------- /agents/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/agents/__init__.py -------------------------------------------------------------------------------- /analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/analysis.py -------------------------------------------------------------------------------- /clean.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/clean.sh -------------------------------------------------------------------------------- /components/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/components/networks.py -------------------------------------------------------------------------------- /components/replay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/components/replay.py -------------------------------------------------------------------------------- /configs/classic_ddqn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/classic_ddqn.json -------------------------------------------------------------------------------- /configs/classic_dqn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/classic_dqn.json -------------------------------------------------------------------------------- /configs/classic_maxmin.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/classic_maxmin.json -------------------------------------------------------------------------------- /configs/lunar_ddqn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/lunar_ddqn.json -------------------------------------------------------------------------------- /configs/lunar_dqn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/lunar_dqn.json -------------------------------------------------------------------------------- /configs/lunar_maxmin.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/lunar_maxmin.json -------------------------------------------------------------------------------- /configs/minatar_ddqn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/minatar_ddqn.json -------------------------------------------------------------------------------- /configs/minatar_dqn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/minatar_dqn.json -------------------------------------------------------------------------------- /configs/minatar_maxmin.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/minatar_maxmin.json -------------------------------------------------------------------------------- /configs/mujoco_ddpg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/mujoco_ddpg.json -------------------------------------------------------------------------------- /configs/mujoco_naf.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/mujoco_naf.json -------------------------------------------------------------------------------- /configs/mujoco_ppo.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/mujoco_ppo.json -------------------------------------------------------------------------------- /configs/mujoco_sac.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/mujoco_sac.json -------------------------------------------------------------------------------- /configs/mujoco_td3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/mujoco_td3.json -------------------------------------------------------------------------------- /configs/nchain_dqn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/nchain_dqn.json -------------------------------------------------------------------------------- /configs/pygame_ddqn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/pygame_ddqn.json -------------------------------------------------------------------------------- /configs/pygame_dqn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/pygame_dqn.json -------------------------------------------------------------------------------- /configs/pygame_maxmin.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/configs/pygame_maxmin.json -------------------------------------------------------------------------------- /envs/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/envs/env.py -------------------------------------------------------------------------------- /envs/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/envs/wrappers.py -------------------------------------------------------------------------------- /experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/experiment.py -------------------------------------------------------------------------------- /find_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/find_config.py -------------------------------------------------------------------------------- /git_commit_id.sh: -------------------------------------------------------------------------------- 1 | git rev-parse --short HEAD -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/main.py -------------------------------------------------------------------------------- /move_log.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/move_log.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/performance.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/results/performance.md -------------------------------------------------------------------------------- /sbatch_m.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/sbatch_m.sh -------------------------------------------------------------------------------- /sbatch_s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/sbatch_s.sh -------------------------------------------------------------------------------- /submit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/submit.py -------------------------------------------------------------------------------- /utils/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/utils/helper.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/plotter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/utils/plotter.py -------------------------------------------------------------------------------- /utils/submitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/utils/submitter.py -------------------------------------------------------------------------------- /utils/sweeper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qlan3/Jaxplorer/HEAD/utils/sweeper.py --------------------------------------------------------------------------------