├── .dockerignore ├── .github └── workflows │ ├── docker-tests.yml │ └── python-publish.yml ├── .gitignore ├── CHANGELOG.md ├── CONTRIBUTING.md ├── Dockerfile ├── LICENSE ├── MANIFEST.in ├── Makefile ├── README.md ├── baselines ├── IPPO │ ├── README.md │ ├── __init__.py │ ├── config │ │ ├── ippo_cnn_overcooked.yaml │ │ ├── ippo_ff_hanabi.yaml │ │ ├── ippo_ff_mabrax.yaml │ │ ├── ippo_ff_mpe.yaml │ │ ├── ippo_ff_mpe_facmac.yaml │ │ ├── ippo_ff_overcooked.yaml │ │ ├── ippo_ff_switch_riddle.yaml │ │ ├── ippo_rnn_hanabi.yaml │ │ ├── ippo_rnn_mpe.yaml │ │ ├── ippo_rnn_overcooked_v2.yaml │ │ └── ippo_rnn_smax.yaml │ ├── ippo_cnn_overcooked.py │ ├── ippo_ff_hanabi.py │ ├── ippo_ff_mabrax.py │ ├── ippo_ff_mpe.py │ ├── ippo_ff_mpe_facmac.py │ ├── ippo_ff_overcooked.py │ ├── ippo_ff_switch_riddle.py │ ├── ippo_rnn_hanabi.py │ ├── ippo_rnn_mpe.py │ ├── ippo_rnn_overcooked_v2.py │ └── ippo_rnn_smax.py ├── MAPPO │ ├── README.md │ ├── config │ │ ├── mappo_homogenous_ff_hanabi.yaml │ │ ├── mappo_homogenous_rnn_hanabi.yaml │ │ ├── mappo_homogenous_rnn_mpe.yaml │ │ └── mappo_homogenous_rnn_smax.yaml │ ├── mappo_ff_hanabi.py │ ├── mappo_rnn_hanabi.py │ ├── mappo_rnn_mpe.py │ └── mappo_rnn_smax.py ├── QLearning │ ├── README.md │ ├── config │ │ ├── alg │ │ │ ├── pqn_vdn_cnn_overcooked.yaml │ │ │ ├── pqn_vdn_ff_hanabi.yaml │ │ │ ├── pqn_vdn_ff_mpe.yaml │ │ │ ├── pqn_vdn_rnn_mpe.yaml │ │ │ ├── pqn_vdn_rnn_smax.yaml │ │ │ ├── ql_cnn_overcooked.yaml │ │ │ ├── ql_rnn_mpe.yaml │ │ │ ├── ql_rnn_smax.yaml │ │ │ ├── shaq_mpe.yaml │ │ │ ├── shaq_smax.yaml │ │ │ ├── transf_qmix_mpe.yaml │ │ │ ├── transf_qmix_smax.yaml │ │ │ └── vdn_ff_mpe.yaml │ │ └── config.yaml │ ├── iql_cnn_overcooked.py │ ├── iql_rnn.py │ ├── pqn_vdn_cnn_overcooked.py │ ├── pqn_vdn_ff.py │ ├── pqn_vdn_rnn.py │ ├── qmix_rnn.py │ ├── shaq.py │ ├── transf_qmix.py │ ├── utils │ │ └── fast_attention.py │ ├── vdn_cnn_overcooked.py │ ├── vdn_ff.py │ └── vdn_rnn.py └── __init__.py ├── docs ├── API │ └── multi_agent_env.md ├── Algorithms │ ├── PPO.md │ └── QLearning.md ├── Environments │ ├── coin_game.md │ ├── hanabi.md │ ├── jaxnav.md │ ├── mabrax.md │ ├── mpe.md │ ├── overcooked.md │ ├── smax.md │ ├── storm.md │ └── switch_riddle.md ├── Installation.md ├── imgs │ ├── coin_game.png │ ├── cramped_room.gif │ ├── hanabi.png │ ├── jaxnav-ma.gif │ ├── mabrax.png │ ├── mpe_qlearning_agent_training-1.png │ ├── mpe_qlearning_speed-1.png │ ├── mpe_speedup-1.png │ ├── qmix_MPE_simple_tag_v3.gif │ ├── sc2_speedup-1.png │ ├── smax.gif │ ├── storm.gif │ ├── storm_nplayer.gif │ └── switch_riddle.png └── index.md ├── jaxmarl ├── __init__.py ├── environments │ ├── __init__.py │ ├── coin_game │ │ ├── README.md │ │ ├── __init__.py │ │ └── coin_game.py │ ├── hanabi │ │ ├── README.md │ │ ├── __init__.py │ │ ├── hanabi.py │ │ ├── hanabi_game.py │ │ ├── manual_game.py │ │ ├── manual_game_human_agents.py │ │ └── pretrained │ │ │ ├── __init__.py │ │ │ ├── download_r2d2_obl.sh │ │ │ ├── obl_r2d2_agent.py │ │ │ └── obl_r2d2_agent_test.py │ ├── jaxnav │ │ ├── README.md │ │ ├── __init__.py │ │ ├── jaxnav_env.py │ │ ├── jaxnav_graph_utils.py │ │ ├── jaxnav_singletons.py │ │ ├── jaxnav_ued_utils.py │ │ ├── jaxnav_utils.py │ │ ├── jaxnav_viz.py │ │ └── maps │ │ │ ├── __init__.py │ │ │ ├── grid_map.py │ │ │ ├── map.py │ │ │ ├── map_registration.py │ │ │ └── polygon_map.py │ ├── mabrax │ │ ├── README.md │ │ ├── __init__.py │ │ ├── mabrax_env.py │ │ └── mappings.py │ ├── mpe │ │ ├── ReadMe.md │ │ ├── __init__.py │ │ ├── default_params.py │ │ ├── mpe_visualizer.py │ │ ├── simple.py │ │ ├── simple_adversary.py │ │ ├── simple_crypto.py │ │ ├── simple_facmac.py │ │ ├── simple_push.py │ │ ├── simple_reference.py │ │ ├── simple_speaker_listener.py │ │ ├── simple_spread.py │ │ ├── simple_tag.py │ │ └── simple_world_comm.py │ ├── multi_agent_env.py │ ├── overcooked │ │ ├── ReadMe.md │ │ ├── __init__.py │ │ ├── common.py │ │ ├── interactive.py │ │ ├── layouts.py │ │ └── overcooked.py │ ├── overcooked_v2 │ │ ├── README.md │ │ ├── __init__.py │ │ ├── common.py │ │ ├── interactive.py │ │ ├── layouts.py │ │ ├── overcooked.py │ │ ├── settings.py │ │ └── utils.py │ ├── smax │ │ ├── README.md │ │ ├── __init__.py │ │ ├── distributions.py │ │ ├── heuristic_enemy.py │ │ ├── heuristic_enemy_smax_env.py │ │ ├── smax_env.py │ │ └── speed.py │ ├── spaces.py │ ├── storm │ │ ├── README.md │ │ ├── __init__.py │ │ ├── rendering.py │ │ ├── storm.py │ │ ├── storm_2p.py │ │ └── storm_env.py │ └── switch_riddle │ │ ├── README.md │ │ ├── __init__.py │ │ └── switch_riddle.py ├── gridworld │ ├── README.md │ ├── __init__.py │ ├── common.py │ ├── env.py │ ├── grid_viz.py │ ├── interactive.py │ ├── ma_maze.py │ ├── maze.py │ └── tabular_q.py ├── registration.py ├── tutorials │ ├── JaxMARL_Walkthrough.ipynb │ ├── mpe_introduction.py │ ├── overcooked_introduction.py │ ├── smax_introduction.py │ ├── storm_2p_introduction.py │ ├── storm_introduction.py │ └── storm_np_introduction.py ├── viz │ ├── __init__.py │ ├── grid_rendering.py │ ├── grid_rendering_v2.py │ ├── overcooked_v2_visualizer.py │ ├── overcooked_visualizer.py │ ├── visualizer.py │ └── window.py └── wrappers │ ├── __init__.py │ ├── baselines.py │ ├── gymnax.py │ └── transformers.py ├── mkdocs.yml ├── pyproject.toml ├── site ├── 404.html ├── API │ └── multi_agent_env │ │ └── index.html ├── Algorithms │ ├── PPO │ │ └── index.html │ └── QLearning │ │ └── index.html ├── Environments │ ├── coin_game │ │ └── index.html │ ├── hanabi │ │ └── index.html │ ├── jaxnav │ │ └── index.html │ ├── mabrax │ │ └── index.html │ ├── mpe │ │ └── index.html │ ├── overcooked │ │ └── index.html │ ├── smax │ │ └── index.html │ ├── storm │ │ └── index.html │ └── switch_riddle │ │ └── index.html ├── Installation │ └── index.html ├── assets │ ├── _mkdocstrings.css │ ├── images │ │ └── favicon.png │ ├── javascripts │ │ ├── bundle.f1b6f286.min.js │ │ ├── bundle.f1b6f286.min.js.map │ │ ├── lunr │ │ │ ├── min │ │ │ │ ├── lunr.ar.min.js │ │ │ │ ├── lunr.da.min.js │ │ │ │ ├── lunr.de.min.js │ │ │ │ ├── lunr.du.min.js │ │ │ │ ├── lunr.el.min.js │ │ │ │ ├── lunr.es.min.js │ │ │ │ ├── lunr.fi.min.js │ │ │ │ ├── lunr.fr.min.js │ │ │ │ ├── lunr.he.min.js │ │ │ │ ├── lunr.hi.min.js │ │ │ │ ├── lunr.hu.min.js │ │ │ │ ├── lunr.hy.min.js │ │ │ │ ├── lunr.it.min.js │ │ │ │ ├── lunr.ja.min.js │ │ │ │ ├── lunr.jp.min.js │ │ │ │ ├── lunr.kn.min.js │ │ │ │ ├── lunr.ko.min.js │ │ │ │ ├── lunr.multi.min.js │ │ │ │ ├── lunr.nl.min.js │ │ │ │ ├── lunr.no.min.js │ │ │ │ ├── lunr.pt.min.js │ │ │ │ ├── lunr.ro.min.js │ │ │ │ ├── lunr.ru.min.js │ │ │ │ ├── lunr.sa.min.js │ │ │ │ ├── lunr.stemmer.support.min.js │ │ │ │ ├── lunr.sv.min.js │ │ │ │ ├── lunr.ta.min.js │ │ │ │ ├── lunr.te.min.js │ │ │ │ ├── lunr.th.min.js │ │ │ │ ├── lunr.tr.min.js │ │ │ │ ├── lunr.vi.min.js │ │ │ │ └── lunr.zh.min.js │ │ │ ├── tinyseg.js │ │ │ └── wordcut.js │ │ └── workers │ │ │ ├── search.f8cc74c7.min.js │ │ │ └── search.f8cc74c7.min.js.map │ └── stylesheets │ │ ├── main.8608ea7d.min.css │ │ ├── main.8608ea7d.min.css.map │ │ ├── palette.06af60db.min.css │ │ └── palette.06af60db.min.css.map ├── imgs │ ├── coin_game.png │ ├── cramped_room.gif │ ├── hanabi.png │ ├── jaxnav-ma.gif │ ├── mabrax.png │ ├── mpe_qlearning_agent_training-1.png │ ├── mpe_qlearning_speed-1.png │ ├── mpe_speedup-1.png │ ├── qmix_MPE_simple_tag_v3.gif │ ├── sc2_speedup-1.png │ ├── smax.gif │ ├── storm.gif │ ├── storm_nplayer.gif │ └── switch_riddle.png ├── index.html ├── objects.inv ├── search │ └── search_index.json ├── sitemap.xml └── sitemap.xml.gz └── tests ├── baselines ├── test_ippo_mabrax.py ├── test_mappo_smax.py ├── test_pqn_overcooked.py └── test_qmix_mpe.py ├── brax └── test_brax_rand_acts.py ├── coin_game └── test_coin_game_rand_acts.py ├── hanabi ├── actions.json ├── decks.json ├── scores.txt └── test_hanabi.py ├── jaxnav ├── test_jaxnav_gridpolymap.py └── test_jaxnav_rand_acts.py ├── mpe ├── _test_utils │ └── rollout_manager.py ├── mpe_policy_transfer.py ├── test_mpe.py └── zoo_benchmark.py ├── overcooked └── test_overcooked_rand_acts.py ├── overcooked_v2 └── test_overcooked_v2_rand_acts.py ├── smax └── test_smax.py ├── storm └── test_storm_rand_acts.py ├── switch_riddle ├── benchmark.py ├── rollout_menager.py ├── switch_riddle_nojax.py └── test_heuristic.py ├── test_envs_implement_base_api.py └── test_jaxmarl_api.py /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/.dockerignore -------------------------------------------------------------------------------- /.github/workflows/docker-tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/.github/workflows/docker-tests.yml -------------------------------------------------------------------------------- /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/README.md -------------------------------------------------------------------------------- /baselines/IPPO/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/README.md -------------------------------------------------------------------------------- /baselines/IPPO/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /baselines/IPPO/config/ippo_cnn_overcooked.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/config/ippo_cnn_overcooked.yaml -------------------------------------------------------------------------------- /baselines/IPPO/config/ippo_ff_hanabi.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/config/ippo_ff_hanabi.yaml -------------------------------------------------------------------------------- /baselines/IPPO/config/ippo_ff_mabrax.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/config/ippo_ff_mabrax.yaml -------------------------------------------------------------------------------- /baselines/IPPO/config/ippo_ff_mpe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/config/ippo_ff_mpe.yaml -------------------------------------------------------------------------------- /baselines/IPPO/config/ippo_ff_mpe_facmac.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/config/ippo_ff_mpe_facmac.yaml -------------------------------------------------------------------------------- /baselines/IPPO/config/ippo_ff_overcooked.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/config/ippo_ff_overcooked.yaml -------------------------------------------------------------------------------- /baselines/IPPO/config/ippo_ff_switch_riddle.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/config/ippo_ff_switch_riddle.yaml -------------------------------------------------------------------------------- /baselines/IPPO/config/ippo_rnn_hanabi.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/config/ippo_rnn_hanabi.yaml -------------------------------------------------------------------------------- /baselines/IPPO/config/ippo_rnn_mpe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/config/ippo_rnn_mpe.yaml -------------------------------------------------------------------------------- /baselines/IPPO/config/ippo_rnn_overcooked_v2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/config/ippo_rnn_overcooked_v2.yaml -------------------------------------------------------------------------------- /baselines/IPPO/config/ippo_rnn_smax.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/config/ippo_rnn_smax.yaml -------------------------------------------------------------------------------- /baselines/IPPO/ippo_cnn_overcooked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/ippo_cnn_overcooked.py -------------------------------------------------------------------------------- /baselines/IPPO/ippo_ff_hanabi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/ippo_ff_hanabi.py -------------------------------------------------------------------------------- /baselines/IPPO/ippo_ff_mabrax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/ippo_ff_mabrax.py -------------------------------------------------------------------------------- /baselines/IPPO/ippo_ff_mpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/ippo_ff_mpe.py -------------------------------------------------------------------------------- /baselines/IPPO/ippo_ff_mpe_facmac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/ippo_ff_mpe_facmac.py -------------------------------------------------------------------------------- /baselines/IPPO/ippo_ff_overcooked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/ippo_ff_overcooked.py -------------------------------------------------------------------------------- /baselines/IPPO/ippo_ff_switch_riddle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/ippo_ff_switch_riddle.py -------------------------------------------------------------------------------- /baselines/IPPO/ippo_rnn_hanabi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/ippo_rnn_hanabi.py -------------------------------------------------------------------------------- /baselines/IPPO/ippo_rnn_mpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/ippo_rnn_mpe.py -------------------------------------------------------------------------------- /baselines/IPPO/ippo_rnn_overcooked_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/ippo_rnn_overcooked_v2.py -------------------------------------------------------------------------------- /baselines/IPPO/ippo_rnn_smax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/IPPO/ippo_rnn_smax.py -------------------------------------------------------------------------------- /baselines/MAPPO/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/MAPPO/README.md -------------------------------------------------------------------------------- /baselines/MAPPO/config/mappo_homogenous_ff_hanabi.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/MAPPO/config/mappo_homogenous_ff_hanabi.yaml -------------------------------------------------------------------------------- /baselines/MAPPO/config/mappo_homogenous_rnn_hanabi.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/MAPPO/config/mappo_homogenous_rnn_hanabi.yaml -------------------------------------------------------------------------------- /baselines/MAPPO/config/mappo_homogenous_rnn_mpe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/MAPPO/config/mappo_homogenous_rnn_mpe.yaml -------------------------------------------------------------------------------- /baselines/MAPPO/config/mappo_homogenous_rnn_smax.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/MAPPO/config/mappo_homogenous_rnn_smax.yaml -------------------------------------------------------------------------------- /baselines/MAPPO/mappo_ff_hanabi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/MAPPO/mappo_ff_hanabi.py -------------------------------------------------------------------------------- /baselines/MAPPO/mappo_rnn_hanabi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/MAPPO/mappo_rnn_hanabi.py -------------------------------------------------------------------------------- /baselines/MAPPO/mappo_rnn_mpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/MAPPO/mappo_rnn_mpe.py -------------------------------------------------------------------------------- /baselines/MAPPO/mappo_rnn_smax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/MAPPO/mappo_rnn_smax.py -------------------------------------------------------------------------------- /baselines/QLearning/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/README.md -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/pqn_vdn_cnn_overcooked.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/pqn_vdn_cnn_overcooked.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/pqn_vdn_ff_hanabi.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/pqn_vdn_ff_hanabi.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/pqn_vdn_ff_mpe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/pqn_vdn_ff_mpe.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/pqn_vdn_rnn_mpe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/pqn_vdn_rnn_mpe.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/pqn_vdn_rnn_smax.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/pqn_vdn_rnn_smax.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/ql_cnn_overcooked.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/ql_cnn_overcooked.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/ql_rnn_mpe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/ql_rnn_mpe.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/ql_rnn_smax.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/ql_rnn_smax.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/shaq_mpe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/shaq_mpe.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/shaq_smax.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/shaq_smax.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/transf_qmix_mpe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/transf_qmix_mpe.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/transf_qmix_smax.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/transf_qmix_smax.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/alg/vdn_ff_mpe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/alg/vdn_ff_mpe.yaml -------------------------------------------------------------------------------- /baselines/QLearning/config/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/config/config.yaml -------------------------------------------------------------------------------- /baselines/QLearning/iql_cnn_overcooked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/iql_cnn_overcooked.py -------------------------------------------------------------------------------- /baselines/QLearning/iql_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/iql_rnn.py -------------------------------------------------------------------------------- /baselines/QLearning/pqn_vdn_cnn_overcooked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/pqn_vdn_cnn_overcooked.py -------------------------------------------------------------------------------- /baselines/QLearning/pqn_vdn_ff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/pqn_vdn_ff.py -------------------------------------------------------------------------------- /baselines/QLearning/pqn_vdn_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/pqn_vdn_rnn.py -------------------------------------------------------------------------------- /baselines/QLearning/qmix_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/qmix_rnn.py -------------------------------------------------------------------------------- /baselines/QLearning/shaq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/shaq.py -------------------------------------------------------------------------------- /baselines/QLearning/transf_qmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/transf_qmix.py -------------------------------------------------------------------------------- /baselines/QLearning/utils/fast_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/utils/fast_attention.py -------------------------------------------------------------------------------- /baselines/QLearning/vdn_cnn_overcooked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/vdn_cnn_overcooked.py -------------------------------------------------------------------------------- /baselines/QLearning/vdn_ff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/vdn_ff.py -------------------------------------------------------------------------------- /baselines/QLearning/vdn_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/baselines/QLearning/vdn_rnn.py -------------------------------------------------------------------------------- /baselines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/API/multi_agent_env.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/API/multi_agent_env.md -------------------------------------------------------------------------------- /docs/Algorithms/PPO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Algorithms/PPO.md -------------------------------------------------------------------------------- /docs/Algorithms/QLearning.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Algorithms/QLearning.md -------------------------------------------------------------------------------- /docs/Environments/coin_game.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Environments/coin_game.md -------------------------------------------------------------------------------- /docs/Environments/hanabi.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Environments/hanabi.md -------------------------------------------------------------------------------- /docs/Environments/jaxnav.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Environments/jaxnav.md -------------------------------------------------------------------------------- /docs/Environments/mabrax.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Environments/mabrax.md -------------------------------------------------------------------------------- /docs/Environments/mpe.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Environments/mpe.md -------------------------------------------------------------------------------- /docs/Environments/overcooked.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Environments/overcooked.md -------------------------------------------------------------------------------- /docs/Environments/smax.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Environments/smax.md -------------------------------------------------------------------------------- /docs/Environments/storm.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Environments/storm.md -------------------------------------------------------------------------------- /docs/Environments/switch_riddle.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Environments/switch_riddle.md -------------------------------------------------------------------------------- /docs/Installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/Installation.md -------------------------------------------------------------------------------- /docs/imgs/coin_game.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/coin_game.png -------------------------------------------------------------------------------- /docs/imgs/cramped_room.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/cramped_room.gif -------------------------------------------------------------------------------- /docs/imgs/hanabi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/hanabi.png -------------------------------------------------------------------------------- /docs/imgs/jaxnav-ma.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/jaxnav-ma.gif -------------------------------------------------------------------------------- /docs/imgs/mabrax.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/mabrax.png -------------------------------------------------------------------------------- /docs/imgs/mpe_qlearning_agent_training-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/mpe_qlearning_agent_training-1.png -------------------------------------------------------------------------------- /docs/imgs/mpe_qlearning_speed-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/mpe_qlearning_speed-1.png -------------------------------------------------------------------------------- /docs/imgs/mpe_speedup-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/mpe_speedup-1.png -------------------------------------------------------------------------------- /docs/imgs/qmix_MPE_simple_tag_v3.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/qmix_MPE_simple_tag_v3.gif -------------------------------------------------------------------------------- /docs/imgs/sc2_speedup-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/sc2_speedup-1.png -------------------------------------------------------------------------------- /docs/imgs/smax.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/smax.gif -------------------------------------------------------------------------------- /docs/imgs/storm.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/storm.gif -------------------------------------------------------------------------------- /docs/imgs/storm_nplayer.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/storm_nplayer.gif -------------------------------------------------------------------------------- /docs/imgs/switch_riddle.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/imgs/switch_riddle.png -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/docs/index.md -------------------------------------------------------------------------------- /jaxmarl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/coin_game/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/coin_game/README.md -------------------------------------------------------------------------------- /jaxmarl/environments/coin_game/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/coin_game/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/coin_game/coin_game.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/coin_game/coin_game.py -------------------------------------------------------------------------------- /jaxmarl/environments/hanabi/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/hanabi/README.md -------------------------------------------------------------------------------- /jaxmarl/environments/hanabi/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/hanabi/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/hanabi/hanabi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/hanabi/hanabi.py -------------------------------------------------------------------------------- /jaxmarl/environments/hanabi/hanabi_game.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/hanabi/hanabi_game.py -------------------------------------------------------------------------------- /jaxmarl/environments/hanabi/manual_game.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/hanabi/manual_game.py -------------------------------------------------------------------------------- /jaxmarl/environments/hanabi/manual_game_human_agents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/hanabi/manual_game_human_agents.py -------------------------------------------------------------------------------- /jaxmarl/environments/hanabi/pretrained/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/hanabi/pretrained/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/hanabi/pretrained/download_r2d2_obl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/hanabi/pretrained/download_r2d2_obl.sh -------------------------------------------------------------------------------- /jaxmarl/environments/hanabi/pretrained/obl_r2d2_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/hanabi/pretrained/obl_r2d2_agent.py -------------------------------------------------------------------------------- /jaxmarl/environments/hanabi/pretrained/obl_r2d2_agent_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/hanabi/pretrained/obl_r2d2_agent_test.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/README.md -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/jaxnav_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/jaxnav_env.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/jaxnav_graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/jaxnav_graph_utils.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/jaxnav_singletons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/jaxnav_singletons.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/jaxnav_ued_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/jaxnav_ued_utils.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/jaxnav_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/jaxnav_utils.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/jaxnav_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/jaxnav_viz.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/maps/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/maps/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/maps/grid_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/maps/grid_map.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/maps/map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/maps/map.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/maps/map_registration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/maps/map_registration.py -------------------------------------------------------------------------------- /jaxmarl/environments/jaxnav/maps/polygon_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/jaxnav/maps/polygon_map.py -------------------------------------------------------------------------------- /jaxmarl/environments/mabrax/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mabrax/README.md -------------------------------------------------------------------------------- /jaxmarl/environments/mabrax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mabrax/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/mabrax/mabrax_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mabrax/mabrax_env.py -------------------------------------------------------------------------------- /jaxmarl/environments/mabrax/mappings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mabrax/mappings.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/ReadMe.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/ReadMe.md -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/default_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/default_params.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/mpe_visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/mpe_visualizer.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/simple.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/simple_adversary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/simple_adversary.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/simple_crypto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/simple_crypto.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/simple_facmac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/simple_facmac.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/simple_push.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/simple_push.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/simple_reference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/simple_reference.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/simple_speaker_listener.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/simple_speaker_listener.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/simple_spread.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/simple_spread.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/simple_tag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/simple_tag.py -------------------------------------------------------------------------------- /jaxmarl/environments/mpe/simple_world_comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/mpe/simple_world_comm.py -------------------------------------------------------------------------------- /jaxmarl/environments/multi_agent_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/multi_agent_env.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked/ReadMe.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked/ReadMe.md -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked/common.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked/interactive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked/interactive.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked/layouts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked/layouts.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked/overcooked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked/overcooked.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked_v2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked_v2/README.md -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked_v2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked_v2/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked_v2/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked_v2/common.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked_v2/interactive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked_v2/interactive.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked_v2/layouts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked_v2/layouts.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked_v2/overcooked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked_v2/overcooked.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked_v2/settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked_v2/settings.py -------------------------------------------------------------------------------- /jaxmarl/environments/overcooked_v2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/overcooked_v2/utils.py -------------------------------------------------------------------------------- /jaxmarl/environments/smax/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/smax/README.md -------------------------------------------------------------------------------- /jaxmarl/environments/smax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/smax/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/smax/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/smax/distributions.py -------------------------------------------------------------------------------- /jaxmarl/environments/smax/heuristic_enemy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/smax/heuristic_enemy.py -------------------------------------------------------------------------------- /jaxmarl/environments/smax/heuristic_enemy_smax_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/smax/heuristic_enemy_smax_env.py -------------------------------------------------------------------------------- /jaxmarl/environments/smax/smax_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/smax/smax_env.py -------------------------------------------------------------------------------- /jaxmarl/environments/smax/speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/smax/speed.py -------------------------------------------------------------------------------- /jaxmarl/environments/spaces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/spaces.py -------------------------------------------------------------------------------- /jaxmarl/environments/storm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/storm/README.md -------------------------------------------------------------------------------- /jaxmarl/environments/storm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/storm/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/storm/rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/storm/rendering.py -------------------------------------------------------------------------------- /jaxmarl/environments/storm/storm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/storm/storm.py -------------------------------------------------------------------------------- /jaxmarl/environments/storm/storm_2p.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/storm/storm_2p.py -------------------------------------------------------------------------------- /jaxmarl/environments/storm/storm_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/storm/storm_env.py -------------------------------------------------------------------------------- /jaxmarl/environments/switch_riddle/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/switch_riddle/README.md -------------------------------------------------------------------------------- /jaxmarl/environments/switch_riddle/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/switch_riddle/__init__.py -------------------------------------------------------------------------------- /jaxmarl/environments/switch_riddle/switch_riddle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/environments/switch_riddle/switch_riddle.py -------------------------------------------------------------------------------- /jaxmarl/gridworld/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/gridworld/README.md -------------------------------------------------------------------------------- /jaxmarl/gridworld/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /jaxmarl/gridworld/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/gridworld/common.py -------------------------------------------------------------------------------- /jaxmarl/gridworld/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/gridworld/env.py -------------------------------------------------------------------------------- /jaxmarl/gridworld/grid_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/gridworld/grid_viz.py -------------------------------------------------------------------------------- /jaxmarl/gridworld/interactive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/gridworld/interactive.py -------------------------------------------------------------------------------- /jaxmarl/gridworld/ma_maze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/gridworld/ma_maze.py -------------------------------------------------------------------------------- /jaxmarl/gridworld/maze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/gridworld/maze.py -------------------------------------------------------------------------------- /jaxmarl/gridworld/tabular_q.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/gridworld/tabular_q.py -------------------------------------------------------------------------------- /jaxmarl/registration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/registration.py -------------------------------------------------------------------------------- /jaxmarl/tutorials/JaxMARL_Walkthrough.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/tutorials/JaxMARL_Walkthrough.ipynb -------------------------------------------------------------------------------- /jaxmarl/tutorials/mpe_introduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/tutorials/mpe_introduction.py -------------------------------------------------------------------------------- /jaxmarl/tutorials/overcooked_introduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/tutorials/overcooked_introduction.py -------------------------------------------------------------------------------- /jaxmarl/tutorials/smax_introduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/tutorials/smax_introduction.py -------------------------------------------------------------------------------- /jaxmarl/tutorials/storm_2p_introduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/tutorials/storm_2p_introduction.py -------------------------------------------------------------------------------- /jaxmarl/tutorials/storm_introduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/tutorials/storm_introduction.py -------------------------------------------------------------------------------- /jaxmarl/tutorials/storm_np_introduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/tutorials/storm_np_introduction.py -------------------------------------------------------------------------------- /jaxmarl/viz/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /jaxmarl/viz/grid_rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/viz/grid_rendering.py -------------------------------------------------------------------------------- /jaxmarl/viz/grid_rendering_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/viz/grid_rendering_v2.py -------------------------------------------------------------------------------- /jaxmarl/viz/overcooked_v2_visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/viz/overcooked_v2_visualizer.py -------------------------------------------------------------------------------- /jaxmarl/viz/overcooked_visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/viz/overcooked_visualizer.py -------------------------------------------------------------------------------- /jaxmarl/viz/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/viz/visualizer.py -------------------------------------------------------------------------------- /jaxmarl/viz/window.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/viz/window.py -------------------------------------------------------------------------------- /jaxmarl/wrappers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /jaxmarl/wrappers/baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/wrappers/baselines.py -------------------------------------------------------------------------------- /jaxmarl/wrappers/gymnax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/wrappers/gymnax.py -------------------------------------------------------------------------------- /jaxmarl/wrappers/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/jaxmarl/wrappers/transformers.py -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/pyproject.toml -------------------------------------------------------------------------------- /site/404.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/404.html -------------------------------------------------------------------------------- /site/API/multi_agent_env/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/API/multi_agent_env/index.html -------------------------------------------------------------------------------- /site/Algorithms/PPO/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Algorithms/PPO/index.html -------------------------------------------------------------------------------- /site/Algorithms/QLearning/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Algorithms/QLearning/index.html -------------------------------------------------------------------------------- /site/Environments/coin_game/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Environments/coin_game/index.html -------------------------------------------------------------------------------- /site/Environments/hanabi/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Environments/hanabi/index.html -------------------------------------------------------------------------------- /site/Environments/jaxnav/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Environments/jaxnav/index.html -------------------------------------------------------------------------------- /site/Environments/mabrax/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Environments/mabrax/index.html -------------------------------------------------------------------------------- /site/Environments/mpe/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Environments/mpe/index.html -------------------------------------------------------------------------------- /site/Environments/overcooked/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Environments/overcooked/index.html -------------------------------------------------------------------------------- /site/Environments/smax/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Environments/smax/index.html -------------------------------------------------------------------------------- /site/Environments/storm/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Environments/storm/index.html -------------------------------------------------------------------------------- /site/Environments/switch_riddle/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Environments/switch_riddle/index.html -------------------------------------------------------------------------------- /site/Installation/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/Installation/index.html -------------------------------------------------------------------------------- /site/assets/_mkdocstrings.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/_mkdocstrings.css -------------------------------------------------------------------------------- /site/assets/images/favicon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/images/favicon.png -------------------------------------------------------------------------------- /site/assets/javascripts/bundle.f1b6f286.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/bundle.f1b6f286.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/bundle.f1b6f286.min.js.map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/bundle.f1b6f286.min.js.map -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.ar.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.ar.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.da.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.da.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.de.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.de.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.du.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.du.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.el.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.el.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.es.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.es.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.fi.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.fi.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.fr.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.fr.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.he.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.he.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.hi.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.hi.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.hu.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.hu.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.hy.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.hy.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.it.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.it.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.ja.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.ja.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.jp.min.js: -------------------------------------------------------------------------------- 1 | module.exports=require("./lunr.ja"); -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.kn.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.kn.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.ko.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.ko.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.multi.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.multi.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.nl.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.nl.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.no.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.no.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.pt.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.pt.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.ro.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.ro.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.ru.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.ru.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.sa.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.sa.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.stemmer.support.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.stemmer.support.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.sv.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.sv.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.ta.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.ta.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.te.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.te.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.th.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.th.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.tr.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.tr.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.vi.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.vi.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/min/lunr.zh.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/min/lunr.zh.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/tinyseg.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/tinyseg.js -------------------------------------------------------------------------------- /site/assets/javascripts/lunr/wordcut.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/lunr/wordcut.js -------------------------------------------------------------------------------- /site/assets/javascripts/workers/search.f8cc74c7.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/workers/search.f8cc74c7.min.js -------------------------------------------------------------------------------- /site/assets/javascripts/workers/search.f8cc74c7.min.js.map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/javascripts/workers/search.f8cc74c7.min.js.map -------------------------------------------------------------------------------- /site/assets/stylesheets/main.8608ea7d.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/stylesheets/main.8608ea7d.min.css -------------------------------------------------------------------------------- /site/assets/stylesheets/main.8608ea7d.min.css.map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/stylesheets/main.8608ea7d.min.css.map -------------------------------------------------------------------------------- /site/assets/stylesheets/palette.06af60db.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/stylesheets/palette.06af60db.min.css -------------------------------------------------------------------------------- /site/assets/stylesheets/palette.06af60db.min.css.map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/assets/stylesheets/palette.06af60db.min.css.map -------------------------------------------------------------------------------- /site/imgs/coin_game.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/coin_game.png -------------------------------------------------------------------------------- /site/imgs/cramped_room.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/cramped_room.gif -------------------------------------------------------------------------------- /site/imgs/hanabi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/hanabi.png -------------------------------------------------------------------------------- /site/imgs/jaxnav-ma.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/jaxnav-ma.gif -------------------------------------------------------------------------------- /site/imgs/mabrax.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/mabrax.png -------------------------------------------------------------------------------- /site/imgs/mpe_qlearning_agent_training-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/mpe_qlearning_agent_training-1.png -------------------------------------------------------------------------------- /site/imgs/mpe_qlearning_speed-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/mpe_qlearning_speed-1.png -------------------------------------------------------------------------------- /site/imgs/mpe_speedup-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/mpe_speedup-1.png -------------------------------------------------------------------------------- /site/imgs/qmix_MPE_simple_tag_v3.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/qmix_MPE_simple_tag_v3.gif -------------------------------------------------------------------------------- /site/imgs/sc2_speedup-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/sc2_speedup-1.png -------------------------------------------------------------------------------- /site/imgs/smax.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/smax.gif -------------------------------------------------------------------------------- /site/imgs/storm.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/storm.gif -------------------------------------------------------------------------------- /site/imgs/storm_nplayer.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/storm_nplayer.gif -------------------------------------------------------------------------------- /site/imgs/switch_riddle.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/imgs/switch_riddle.png -------------------------------------------------------------------------------- /site/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/index.html -------------------------------------------------------------------------------- /site/objects.inv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/objects.inv -------------------------------------------------------------------------------- /site/search/search_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/search/search_index.json -------------------------------------------------------------------------------- /site/sitemap.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/sitemap.xml -------------------------------------------------------------------------------- /site/sitemap.xml.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/site/sitemap.xml.gz -------------------------------------------------------------------------------- /tests/baselines/test_ippo_mabrax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/baselines/test_ippo_mabrax.py -------------------------------------------------------------------------------- /tests/baselines/test_mappo_smax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/baselines/test_mappo_smax.py -------------------------------------------------------------------------------- /tests/baselines/test_pqn_overcooked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/baselines/test_pqn_overcooked.py -------------------------------------------------------------------------------- /tests/baselines/test_qmix_mpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/baselines/test_qmix_mpe.py -------------------------------------------------------------------------------- /tests/brax/test_brax_rand_acts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/brax/test_brax_rand_acts.py -------------------------------------------------------------------------------- /tests/coin_game/test_coin_game_rand_acts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/coin_game/test_coin_game_rand_acts.py -------------------------------------------------------------------------------- /tests/hanabi/actions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/hanabi/actions.json -------------------------------------------------------------------------------- /tests/hanabi/decks.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/hanabi/decks.json -------------------------------------------------------------------------------- /tests/hanabi/scores.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/hanabi/scores.txt -------------------------------------------------------------------------------- /tests/hanabi/test_hanabi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/hanabi/test_hanabi.py -------------------------------------------------------------------------------- /tests/jaxnav/test_jaxnav_gridpolymap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/jaxnav/test_jaxnav_gridpolymap.py -------------------------------------------------------------------------------- /tests/jaxnav/test_jaxnav_rand_acts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/jaxnav/test_jaxnav_rand_acts.py -------------------------------------------------------------------------------- /tests/mpe/_test_utils/rollout_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/mpe/_test_utils/rollout_manager.py -------------------------------------------------------------------------------- /tests/mpe/mpe_policy_transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/mpe/mpe_policy_transfer.py -------------------------------------------------------------------------------- /tests/mpe/test_mpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/mpe/test_mpe.py -------------------------------------------------------------------------------- /tests/mpe/zoo_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/mpe/zoo_benchmark.py -------------------------------------------------------------------------------- /tests/overcooked/test_overcooked_rand_acts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/overcooked/test_overcooked_rand_acts.py -------------------------------------------------------------------------------- /tests/overcooked_v2/test_overcooked_v2_rand_acts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/overcooked_v2/test_overcooked_v2_rand_acts.py -------------------------------------------------------------------------------- /tests/smax/test_smax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/smax/test_smax.py -------------------------------------------------------------------------------- /tests/storm/test_storm_rand_acts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/storm/test_storm_rand_acts.py -------------------------------------------------------------------------------- /tests/switch_riddle/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/switch_riddle/benchmark.py -------------------------------------------------------------------------------- /tests/switch_riddle/rollout_menager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/switch_riddle/rollout_menager.py -------------------------------------------------------------------------------- /tests/switch_riddle/switch_riddle_nojax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/switch_riddle/switch_riddle_nojax.py -------------------------------------------------------------------------------- /tests/switch_riddle/test_heuristic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/switch_riddle/test_heuristic.py -------------------------------------------------------------------------------- /tests/test_envs_implement_base_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/test_envs_implement_base_api.py -------------------------------------------------------------------------------- /tests/test_jaxmarl_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FLAIROx/JaxMARL/HEAD/tests/test_jaxmarl_api.py --------------------------------------------------------------------------------