├── .github ├── CONTRIBUTING.md ├── pull_request_template.md └── workflows │ └── ci.yaml ├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yaml ├── LICENSE ├── README.md ├── docs ├── api_documentation │ ├── core │ │ ├── aurora.md │ │ ├── cma_mega.md │ │ ├── cmaes.md │ │ ├── cmame.md │ │ ├── containers.md │ │ ├── dads.md │ │ ├── dcrlme.md │ │ ├── diayn.md │ │ ├── emitters.md │ │ ├── genetic_algorithm.md │ │ ├── map_elites.md │ │ ├── me_pbt.md │ │ ├── mees.md │ │ ├── mels.md │ │ ├── mome.md │ │ ├── neuroevolution.md │ │ ├── nsga2.md │ │ ├── omg_mega.md │ │ ├── pbt.md │ │ ├── pga_aurora.md │ │ ├── pgame.md │ │ ├── qdpg.md │ │ ├── sac.md │ │ ├── smerl.md │ │ ├── spea2.md │ │ └── td3.md │ ├── tasks.md │ └── utils.md ├── caveats.md ├── css │ └── mkdocstrings.css ├── examples ├── guides │ └── CONTRIBUTING.md ├── img │ ├── AIRL_logo.png │ ├── favicon.ico │ ├── instadeep_logo.png │ └── qdax_logo.png ├── index.md ├── installation.md └── overview.md ├── examples ├── aurora.ipynb ├── cmaes.ipynb ├── cmame.ipynb ├── cmamega.ipynb ├── dads.ipynb ├── dcrlme.ipynb ├── diayn.ipynb ├── distributed_mapelites.ipynb ├── dns.ipynb ├── jumanji_snake.ipynb ├── mapelites.ipynb ├── mapelites_asktell.ipynb ├── me_sac_pbt.ipynb ├── me_td3_pbt.ipynb ├── mees.ipynb ├── mels.ipynb ├── mome.ipynb ├── nsga2_spea2.ipynb ├── omgmega.ipynb ├── pga_aurora.ipynb ├── pgame.ipynb ├── qdpg.ipynb ├── sac_pbt.ipynb ├── smerl.ipynb └── td3_pbt.ipynb ├── mkdocs.yaml ├── pyproject.toml ├── qdax ├── __init__.py ├── baselines │ ├── __init__.py │ ├── cmaes.py │ ├── dads.py │ ├── dads_smerl.py │ ├── diayn.py │ ├── diayn_smerl.py │ ├── genetic_algorithm.py │ ├── nsga2.py │ ├── pbt.py │ ├── sac.py │ ├── sac_pbt.py │ ├── spea2.py │ ├── td3.py │ └── td3_pbt.py ├── core │ ├── __init__.py │ ├── aurora.py │ ├── containers │ │ ├── __init__.py │ │ ├── archive.py │ │ ├── dns_repertoire.py │ │ ├── ga_repertoire.py │ │ ├── mapelites_repertoire.py │ │ ├── mels_repertoire.py │ │ ├── mome_repertoire.py │ │ ├── nsga2_repertoire.py │ │ ├── repertoire.py │ │ ├── spea2_repertoire.py │ │ ├── uniform_replacement_archive.py │ │ └── unstructured_repertoire.py │ ├── distributed_map_elites.py │ ├── dns.py │ ├── emitters │ │ ├── __init__.py │ │ ├── cma_emitter.py │ │ ├── cma_improvement_emitter.py │ │ ├── cma_mega_emitter.py │ │ ├── cma_opt_emitter.py │ │ ├── cma_pool_emitter.py │ │ ├── cma_rnd_emitter.py │ │ ├── dcrl_emitter.py │ │ ├── dcrl_me_emitter.py │ │ ├── dpg_emitter.py │ │ ├── emitter.py │ │ ├── mees_emitter.py │ │ ├── multi_emitter.py │ │ ├── mutation_operators.py │ │ ├── omg_mega_emitter.py │ │ ├── pbt_me_emitter.py │ │ ├── pbt_variation_operators.py │ │ ├── pga_me_emitter.py │ │ ├── qdpg_emitter.py │ │ ├── qpg_emitter.py │ │ ├── repertoire_selectors │ │ │ ├── __init__.py │ │ │ ├── mome_uniform_selector.py │ │ │ ├── selector.py │ │ │ └── uniform_selector.py │ │ └── standard_emitters.py │ ├── map_elites.py │ ├── mels.py │ ├── mome.py │ └── neuroevolution │ │ ├── __init__.py │ │ ├── buffers │ │ ├── __init__.py │ │ ├── buffer.py │ │ └── trajectory_buffer.py │ │ ├── losses │ │ ├── __init__.py │ │ ├── dads_loss.py │ │ ├── diayn_loss.py │ │ ├── sac_loss.py │ │ └── td3_loss.py │ │ ├── mdp_utils.py │ │ ├── networks │ │ ├── __init__.py │ │ ├── dads_networks.py │ │ ├── diayn_networks.py │ │ ├── networks.py │ │ ├── sac_networks.py │ │ ├── seq2seq_networks.py │ │ └── td3_networks.py │ │ ├── normalization_utils.py │ │ └── sac_td3_utils.py ├── custom_types.py ├── tasks │ ├── README.md │ ├── __init__.py │ ├── arm.py │ ├── brax │ │ ├── __init__.py │ │ ├── descriptor_extractors.py │ │ ├── env_creators.py │ │ ├── envs │ │ │ ├── __init__.py │ │ │ └── base_env.py │ │ └── wrappers │ │ │ ├── __init__.py │ │ │ ├── base_wrappers.py │ │ │ ├── eval_metrics_wrapper.py │ │ │ ├── init_state_wrapper.py │ │ │ ├── locomotion_wrappers.py │ │ │ └── reward_wrappers.py │ ├── hypervolume_functions.py │ ├── jumanji_envs.py │ ├── qd_suite │ │ ├── __init__.py │ │ ├── archimedean_spiral.py │ │ ├── deceptive_evolvability.py │ │ ├── qd_suite_task.py │ │ └── ssf.py │ └── standard_functions.py └── utils │ ├── __init__.py │ ├── metrics.py │ ├── pareto_front.py │ ├── plotting.py │ ├── sampling.py │ ├── train_seq2seq.py │ └── uncertainty_metrics.py ├── tests ├── baselines_test │ ├── cmaes_test.py │ ├── cmame_test.py │ ├── cmamega_test.py │ ├── dads_smerl_test.py │ ├── dads_test.py │ ├── dcrlme_test.py │ ├── diayn_smerl_test.py │ ├── diayn_test.py │ ├── ga_test.py │ ├── me_pbt_sac_test.py │ ├── me_pbt_td3_test.py │ ├── mees_test.py │ ├── omgmega_test.py │ ├── pbt_sac_test.py │ ├── pbt_td3_test.py │ ├── pgame_test.py │ ├── qdpg_test.py │ ├── sac_test.py │ └── td3_test.py ├── core_test │ ├── aurora_test.py │ ├── containers_test │ │ ├── archive_test.py │ │ ├── mapelites_repertoire_test.py │ │ └── mels_repertoire_test.py │ ├── dns_test.py │ ├── emitters_test │ │ └── multi_emitter_test.py │ ├── map_elites_test.py │ ├── mels_test.py │ ├── mome_test.py │ └── neuroevolution_test │ │ └── buffers_test │ │ ├── buffer_test.py │ │ └── trajectory_buffer_test.py ├── tasks_test │ ├── arm_test.py │ ├── brax_task_test.py │ ├── hypervolume_functions_test.py │ ├── jumanji_envs_test.py │ ├── qd_suite_test.py │ └── standard_functions_test.py └── utils_test │ ├── metrics_test.py │ ├── plotting_test.py │ ├── sampling_test.py │ └── uncertainty_metrics_test.py └── uv.lock /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.github/workflows/ci.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/.github/workflows/ci.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/README.md -------------------------------------------------------------------------------- /docs/api_documentation/core/aurora.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/aurora.md -------------------------------------------------------------------------------- /docs/api_documentation/core/cma_mega.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/cma_mega.md -------------------------------------------------------------------------------- /docs/api_documentation/core/cmaes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/cmaes.md -------------------------------------------------------------------------------- /docs/api_documentation/core/cmame.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/cmame.md -------------------------------------------------------------------------------- /docs/api_documentation/core/containers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/containers.md -------------------------------------------------------------------------------- /docs/api_documentation/core/dads.md: -------------------------------------------------------------------------------- 1 | # DADS class 2 | 3 | ::: qdax.baselines.dads.DADS 4 | -------------------------------------------------------------------------------- /docs/api_documentation/core/dcrlme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/dcrlme.md -------------------------------------------------------------------------------- /docs/api_documentation/core/diayn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/diayn.md -------------------------------------------------------------------------------- /docs/api_documentation/core/emitters.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/emitters.md -------------------------------------------------------------------------------- /docs/api_documentation/core/genetic_algorithm.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/genetic_algorithm.md -------------------------------------------------------------------------------- /docs/api_documentation/core/map_elites.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/map_elites.md -------------------------------------------------------------------------------- /docs/api_documentation/core/me_pbt.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/me_pbt.md -------------------------------------------------------------------------------- /docs/api_documentation/core/mees.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/mees.md -------------------------------------------------------------------------------- /docs/api_documentation/core/mels.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/mels.md -------------------------------------------------------------------------------- /docs/api_documentation/core/mome.md: -------------------------------------------------------------------------------- 1 | # MOME class 2 | 3 | ::: qdax.core.mome.MOME 4 | -------------------------------------------------------------------------------- /docs/api_documentation/core/neuroevolution.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/neuroevolution.md -------------------------------------------------------------------------------- /docs/api_documentation/core/nsga2.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/nsga2.md -------------------------------------------------------------------------------- /docs/api_documentation/core/omg_mega.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/omg_mega.md -------------------------------------------------------------------------------- /docs/api_documentation/core/pbt.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/pbt.md -------------------------------------------------------------------------------- /docs/api_documentation/core/pga_aurora.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/pga_aurora.md -------------------------------------------------------------------------------- /docs/api_documentation/core/pgame.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/pgame.md -------------------------------------------------------------------------------- /docs/api_documentation/core/qdpg.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/qdpg.md -------------------------------------------------------------------------------- /docs/api_documentation/core/sac.md: -------------------------------------------------------------------------------- 1 | # SAC class 2 | 3 | ::: qdax.baselines.sac.SAC 4 | -------------------------------------------------------------------------------- /docs/api_documentation/core/smerl.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/smerl.md -------------------------------------------------------------------------------- /docs/api_documentation/core/spea2.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/core/spea2.md -------------------------------------------------------------------------------- /docs/api_documentation/core/td3.md: -------------------------------------------------------------------------------- 1 | # TD3 class 2 | 3 | ::: qdax.baselines.td3.TD3 4 | -------------------------------------------------------------------------------- /docs/api_documentation/tasks.md: -------------------------------------------------------------------------------- 1 | # Tasks 2 | 3 | --8<-- "qdax/tasks/README.md" 4 | -------------------------------------------------------------------------------- /docs/api_documentation/utils.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/api_documentation/utils.md -------------------------------------------------------------------------------- /docs/caveats.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/caveats.md -------------------------------------------------------------------------------- /docs/css/mkdocstrings.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/css/mkdocstrings.css -------------------------------------------------------------------------------- /docs/examples: -------------------------------------------------------------------------------- 1 | ../examples/ -------------------------------------------------------------------------------- /docs/guides/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/guides/CONTRIBUTING.md -------------------------------------------------------------------------------- /docs/img/AIRL_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/img/AIRL_logo.png -------------------------------------------------------------------------------- /docs/img/favicon.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/img/favicon.ico -------------------------------------------------------------------------------- /docs/img/instadeep_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/img/instadeep_logo.png -------------------------------------------------------------------------------- /docs/img/qdax_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/img/qdax_logo.png -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- 1 | 2 | --8<-- "README.md" 3 | -------------------------------------------------------------------------------- /docs/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/installation.md -------------------------------------------------------------------------------- /docs/overview.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/docs/overview.md -------------------------------------------------------------------------------- /examples/aurora.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/aurora.ipynb -------------------------------------------------------------------------------- /examples/cmaes.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/cmaes.ipynb -------------------------------------------------------------------------------- /examples/cmame.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/cmame.ipynb -------------------------------------------------------------------------------- /examples/cmamega.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/cmamega.ipynb -------------------------------------------------------------------------------- /examples/dads.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/dads.ipynb -------------------------------------------------------------------------------- /examples/dcrlme.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/dcrlme.ipynb -------------------------------------------------------------------------------- /examples/diayn.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/diayn.ipynb -------------------------------------------------------------------------------- /examples/distributed_mapelites.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/distributed_mapelites.ipynb -------------------------------------------------------------------------------- /examples/dns.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/dns.ipynb -------------------------------------------------------------------------------- /examples/jumanji_snake.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/jumanji_snake.ipynb -------------------------------------------------------------------------------- /examples/mapelites.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/mapelites.ipynb -------------------------------------------------------------------------------- /examples/mapelites_asktell.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/mapelites_asktell.ipynb -------------------------------------------------------------------------------- /examples/me_sac_pbt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/me_sac_pbt.ipynb -------------------------------------------------------------------------------- /examples/me_td3_pbt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/me_td3_pbt.ipynb -------------------------------------------------------------------------------- /examples/mees.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/mees.ipynb -------------------------------------------------------------------------------- /examples/mels.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/mels.ipynb -------------------------------------------------------------------------------- /examples/mome.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/mome.ipynb -------------------------------------------------------------------------------- /examples/nsga2_spea2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/nsga2_spea2.ipynb -------------------------------------------------------------------------------- /examples/omgmega.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/omgmega.ipynb -------------------------------------------------------------------------------- /examples/pga_aurora.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/pga_aurora.ipynb -------------------------------------------------------------------------------- /examples/pgame.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/pgame.ipynb -------------------------------------------------------------------------------- /examples/qdpg.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/qdpg.ipynb -------------------------------------------------------------------------------- /examples/sac_pbt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/sac_pbt.ipynb -------------------------------------------------------------------------------- /examples/smerl.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/smerl.ipynb -------------------------------------------------------------------------------- /examples/td3_pbt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/examples/td3_pbt.ipynb -------------------------------------------------------------------------------- /mkdocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/mkdocs.yaml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/pyproject.toml -------------------------------------------------------------------------------- /qdax/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.5.1" 2 | -------------------------------------------------------------------------------- /qdax/baselines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/baselines/cmaes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/cmaes.py -------------------------------------------------------------------------------- /qdax/baselines/dads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/dads.py -------------------------------------------------------------------------------- /qdax/baselines/dads_smerl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/dads_smerl.py -------------------------------------------------------------------------------- /qdax/baselines/diayn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/diayn.py -------------------------------------------------------------------------------- /qdax/baselines/diayn_smerl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/diayn_smerl.py -------------------------------------------------------------------------------- /qdax/baselines/genetic_algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/genetic_algorithm.py -------------------------------------------------------------------------------- /qdax/baselines/nsga2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/nsga2.py -------------------------------------------------------------------------------- /qdax/baselines/pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/pbt.py -------------------------------------------------------------------------------- /qdax/baselines/sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/sac.py -------------------------------------------------------------------------------- /qdax/baselines/sac_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/sac_pbt.py -------------------------------------------------------------------------------- /qdax/baselines/spea2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/spea2.py -------------------------------------------------------------------------------- /qdax/baselines/td3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/td3.py -------------------------------------------------------------------------------- /qdax/baselines/td3_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/baselines/td3_pbt.py -------------------------------------------------------------------------------- /qdax/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/core/aurora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/aurora.py -------------------------------------------------------------------------------- /qdax/core/containers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/__init__.py -------------------------------------------------------------------------------- /qdax/core/containers/archive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/archive.py -------------------------------------------------------------------------------- /qdax/core/containers/dns_repertoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/dns_repertoire.py -------------------------------------------------------------------------------- /qdax/core/containers/ga_repertoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/ga_repertoire.py -------------------------------------------------------------------------------- /qdax/core/containers/mapelites_repertoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/mapelites_repertoire.py -------------------------------------------------------------------------------- /qdax/core/containers/mels_repertoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/mels_repertoire.py -------------------------------------------------------------------------------- /qdax/core/containers/mome_repertoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/mome_repertoire.py -------------------------------------------------------------------------------- /qdax/core/containers/nsga2_repertoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/nsga2_repertoire.py -------------------------------------------------------------------------------- /qdax/core/containers/repertoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/repertoire.py -------------------------------------------------------------------------------- /qdax/core/containers/spea2_repertoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/spea2_repertoire.py -------------------------------------------------------------------------------- /qdax/core/containers/uniform_replacement_archive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/uniform_replacement_archive.py -------------------------------------------------------------------------------- /qdax/core/containers/unstructured_repertoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/containers/unstructured_repertoire.py -------------------------------------------------------------------------------- /qdax/core/distributed_map_elites.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/distributed_map_elites.py -------------------------------------------------------------------------------- /qdax/core/dns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/dns.py -------------------------------------------------------------------------------- /qdax/core/emitters/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/core/emitters/cma_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/cma_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/cma_improvement_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/cma_improvement_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/cma_mega_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/cma_mega_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/cma_opt_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/cma_opt_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/cma_pool_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/cma_pool_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/cma_rnd_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/cma_rnd_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/dcrl_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/dcrl_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/dcrl_me_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/dcrl_me_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/dpg_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/dpg_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/mees_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/mees_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/multi_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/multi_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/mutation_operators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/mutation_operators.py -------------------------------------------------------------------------------- /qdax/core/emitters/omg_mega_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/omg_mega_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/pbt_me_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/pbt_me_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/pbt_variation_operators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/pbt_variation_operators.py -------------------------------------------------------------------------------- /qdax/core/emitters/pga_me_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/pga_me_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/qdpg_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/qdpg_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/qpg_emitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/qpg_emitter.py -------------------------------------------------------------------------------- /qdax/core/emitters/repertoire_selectors/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/core/emitters/repertoire_selectors/mome_uniform_selector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/repertoire_selectors/mome_uniform_selector.py -------------------------------------------------------------------------------- /qdax/core/emitters/repertoire_selectors/selector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/repertoire_selectors/selector.py -------------------------------------------------------------------------------- /qdax/core/emitters/repertoire_selectors/uniform_selector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/repertoire_selectors/uniform_selector.py -------------------------------------------------------------------------------- /qdax/core/emitters/standard_emitters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/emitters/standard_emitters.py -------------------------------------------------------------------------------- /qdax/core/map_elites.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/map_elites.py -------------------------------------------------------------------------------- /qdax/core/mels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/mels.py -------------------------------------------------------------------------------- /qdax/core/mome.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/mome.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/core/neuroevolution/buffers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/core/neuroevolution/buffers/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/buffers/buffer.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/buffers/trajectory_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/buffers/trajectory_buffer.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/core/neuroevolution/losses/dads_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/losses/dads_loss.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/losses/diayn_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/losses/diayn_loss.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/losses/sac_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/losses/sac_loss.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/losses/td3_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/losses/td3_loss.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/mdp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/mdp_utils.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/networks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/core/neuroevolution/networks/dads_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/networks/dads_networks.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/networks/diayn_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/networks/diayn_networks.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/networks/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/networks/networks.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/networks/sac_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/networks/sac_networks.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/networks/seq2seq_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/networks/seq2seq_networks.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/networks/td3_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/networks/td3_networks.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/normalization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/normalization_utils.py -------------------------------------------------------------------------------- /qdax/core/neuroevolution/sac_td3_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/core/neuroevolution/sac_td3_utils.py -------------------------------------------------------------------------------- /qdax/custom_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/custom_types.py -------------------------------------------------------------------------------- /qdax/tasks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/README.md -------------------------------------------------------------------------------- /qdax/tasks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/tasks/arm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/arm.py -------------------------------------------------------------------------------- /qdax/tasks/brax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/brax/__init__.py -------------------------------------------------------------------------------- /qdax/tasks/brax/descriptor_extractors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/brax/descriptor_extractors.py -------------------------------------------------------------------------------- /qdax/tasks/brax/env_creators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/brax/env_creators.py -------------------------------------------------------------------------------- /qdax/tasks/brax/envs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/tasks/brax/envs/base_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/brax/envs/base_env.py -------------------------------------------------------------------------------- /qdax/tasks/brax/wrappers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/tasks/brax/wrappers/base_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/brax/wrappers/base_wrappers.py -------------------------------------------------------------------------------- /qdax/tasks/brax/wrappers/eval_metrics_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/brax/wrappers/eval_metrics_wrapper.py -------------------------------------------------------------------------------- /qdax/tasks/brax/wrappers/init_state_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/brax/wrappers/init_state_wrapper.py -------------------------------------------------------------------------------- /qdax/tasks/brax/wrappers/locomotion_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/brax/wrappers/locomotion_wrappers.py -------------------------------------------------------------------------------- /qdax/tasks/brax/wrappers/reward_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/brax/wrappers/reward_wrappers.py -------------------------------------------------------------------------------- /qdax/tasks/hypervolume_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/hypervolume_functions.py -------------------------------------------------------------------------------- /qdax/tasks/jumanji_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/jumanji_envs.py -------------------------------------------------------------------------------- /qdax/tasks/qd_suite/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/qd_suite/__init__.py -------------------------------------------------------------------------------- /qdax/tasks/qd_suite/archimedean_spiral.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/qd_suite/archimedean_spiral.py -------------------------------------------------------------------------------- /qdax/tasks/qd_suite/deceptive_evolvability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/qd_suite/deceptive_evolvability.py -------------------------------------------------------------------------------- /qdax/tasks/qd_suite/qd_suite_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/qd_suite/qd_suite_task.py -------------------------------------------------------------------------------- /qdax/tasks/qd_suite/ssf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/qd_suite/ssf.py -------------------------------------------------------------------------------- /qdax/tasks/standard_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/tasks/standard_functions.py -------------------------------------------------------------------------------- /qdax/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /qdax/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/utils/metrics.py -------------------------------------------------------------------------------- /qdax/utils/pareto_front.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/utils/pareto_front.py -------------------------------------------------------------------------------- /qdax/utils/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/utils/plotting.py -------------------------------------------------------------------------------- /qdax/utils/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/utils/sampling.py -------------------------------------------------------------------------------- /qdax/utils/train_seq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/utils/train_seq2seq.py -------------------------------------------------------------------------------- /qdax/utils/uncertainty_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/qdax/utils/uncertainty_metrics.py -------------------------------------------------------------------------------- /tests/baselines_test/cmaes_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/cmaes_test.py -------------------------------------------------------------------------------- /tests/baselines_test/cmame_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/cmame_test.py -------------------------------------------------------------------------------- /tests/baselines_test/cmamega_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/cmamega_test.py -------------------------------------------------------------------------------- /tests/baselines_test/dads_smerl_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/dads_smerl_test.py -------------------------------------------------------------------------------- /tests/baselines_test/dads_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/dads_test.py -------------------------------------------------------------------------------- /tests/baselines_test/dcrlme_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/dcrlme_test.py -------------------------------------------------------------------------------- /tests/baselines_test/diayn_smerl_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/diayn_smerl_test.py -------------------------------------------------------------------------------- /tests/baselines_test/diayn_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/diayn_test.py -------------------------------------------------------------------------------- /tests/baselines_test/ga_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/ga_test.py -------------------------------------------------------------------------------- /tests/baselines_test/me_pbt_sac_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/me_pbt_sac_test.py -------------------------------------------------------------------------------- /tests/baselines_test/me_pbt_td3_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/me_pbt_td3_test.py -------------------------------------------------------------------------------- /tests/baselines_test/mees_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/mees_test.py -------------------------------------------------------------------------------- /tests/baselines_test/omgmega_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/omgmega_test.py -------------------------------------------------------------------------------- /tests/baselines_test/pbt_sac_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/pbt_sac_test.py -------------------------------------------------------------------------------- /tests/baselines_test/pbt_td3_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/pbt_td3_test.py -------------------------------------------------------------------------------- /tests/baselines_test/pgame_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/pgame_test.py -------------------------------------------------------------------------------- /tests/baselines_test/qdpg_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/qdpg_test.py -------------------------------------------------------------------------------- /tests/baselines_test/sac_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/sac_test.py -------------------------------------------------------------------------------- /tests/baselines_test/td3_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/baselines_test/td3_test.py -------------------------------------------------------------------------------- /tests/core_test/aurora_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/core_test/aurora_test.py -------------------------------------------------------------------------------- /tests/core_test/containers_test/archive_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/core_test/containers_test/archive_test.py -------------------------------------------------------------------------------- /tests/core_test/containers_test/mapelites_repertoire_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/core_test/containers_test/mapelites_repertoire_test.py -------------------------------------------------------------------------------- /tests/core_test/containers_test/mels_repertoire_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/core_test/containers_test/mels_repertoire_test.py -------------------------------------------------------------------------------- /tests/core_test/dns_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/core_test/dns_test.py -------------------------------------------------------------------------------- /tests/core_test/emitters_test/multi_emitter_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/core_test/emitters_test/multi_emitter_test.py -------------------------------------------------------------------------------- /tests/core_test/map_elites_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/core_test/map_elites_test.py -------------------------------------------------------------------------------- /tests/core_test/mels_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/core_test/mels_test.py -------------------------------------------------------------------------------- /tests/core_test/mome_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/core_test/mome_test.py -------------------------------------------------------------------------------- /tests/core_test/neuroevolution_test/buffers_test/buffer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/core_test/neuroevolution_test/buffers_test/buffer_test.py -------------------------------------------------------------------------------- /tests/core_test/neuroevolution_test/buffers_test/trajectory_buffer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/core_test/neuroevolution_test/buffers_test/trajectory_buffer_test.py -------------------------------------------------------------------------------- /tests/tasks_test/arm_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/tasks_test/arm_test.py -------------------------------------------------------------------------------- /tests/tasks_test/brax_task_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/tasks_test/brax_task_test.py -------------------------------------------------------------------------------- /tests/tasks_test/hypervolume_functions_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/tasks_test/hypervolume_functions_test.py -------------------------------------------------------------------------------- /tests/tasks_test/jumanji_envs_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/tasks_test/jumanji_envs_test.py -------------------------------------------------------------------------------- /tests/tasks_test/qd_suite_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/tasks_test/qd_suite_test.py -------------------------------------------------------------------------------- /tests/tasks_test/standard_functions_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/tasks_test/standard_functions_test.py -------------------------------------------------------------------------------- /tests/utils_test/metrics_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/utils_test/metrics_test.py -------------------------------------------------------------------------------- /tests/utils_test/plotting_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/utils_test/plotting_test.py -------------------------------------------------------------------------------- /tests/utils_test/sampling_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/utils_test/sampling_test.py -------------------------------------------------------------------------------- /tests/utils_test/uncertainty_metrics_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/tests/utils_test/uncertainty_metrics_test.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adaptive-intelligent-robotics/QDax/HEAD/uv.lock --------------------------------------------------------------------------------