Qdax

What's changed

- Fix dependencies in notebook examples

What's Changed
- minimal support for jumanji 136
- PBT and PBT-ME 143
- Humanoid Trap 143


**Full Changelog**: https://github.com/adaptive-intelligent-robotics/QDax/compare/v0.2.1...v0.2.2

Small release to fix a dependency that was preventing the installation of qdax through pypi. The issue was related to scikit-learn installation. See 130

Moving to v0.2.0 with several new algorithms, fixes of current implementations, a whole new set of tasks, a more general API and some dependencies update to stay in phase with the Jax community :rocket: :fast_forward: :fast_forward:

- fix(envs): order of wrappers to ensure update of state descriptor when using fixed init state (128)
- fix(doc): add colab links, missing doc, update version (125)
- feat!(repertoire): optional extra-scores for repertoire addition (118)
- fix(jit): avoid consecutive jits of same method in for loops (122)
- fix(docker): fix run-image docker stage (121)
- feat(algorithms): add MAP-Elites distributed on multiple devices (117)
- fix(test): inverse fitness and desc names in sampling test (119)
- feat(github): add GitHub template for PR (120)
- feat(algorithms): add QDPG emitter + refactor PGAME (110)
- feat(algorithms): add CMA-ME, fix CMA-ES and CMA-MEGA (86)
- fix(mees): add batch size property (114)
- feat(algorithms): Add Multi-Emitter (90)
- fix: reset_based scoring in brax_env default task (109)
- feat(algorithms): add ME-ES to QDax (81)
- fix(examples): brax version in colab examples (108)
- fix(docs): avoid using flax 0.6.2 in setup (112)
- feat(envs): wrapper for fixed initial state of environments (92)
- fix(style): mypy issue in controller training
- fix: optimizer state reinitialization for PG variations (104)
- fix: add update policy delay in PG emitter
- fix(pointmaze): scale after the clip of actions (101)
- hotfix(images): re-add deleted logos to the repo
- docs: add caveats and logo (99)
- feat: Default Scoring Functions for Sphere, Rastrigin, Arm, Brax environments, Hypervolume functions and QD Suite (73)
- chore: Update jax, brax and flax versions (fixes the jax.tree_util warnings) (76)
- doc: Add remark for installing QDax with GPU support in README (77)
- fix the replay buffer overflow issue (75)
- fix: correct irrelevant factor 0.25 in td3 loss (78)

We also welcome a new contributor: maxencefaldor :clap:

Well done to all the team for this new release :tada: :woman_technologist: :man_technologist: (manon-but-yes Lookatator limbryan ranzenTom Aneoshun Egiob valentinmace maxiallard felixchalumeau)

Finally moving to version 0.1.0 of QDax with several algorithms additions, new tools, enhanced documentation and several fixes :rocket:

- Update notebooks to run on colab and update chex version (61)
- Pin brax version + remove autoreload (60)”
- Enhance the current documentation + gather baselines in separate folder (57)
- Add example notebook for NSGA2 and SPEA2 (58)
- Add API documentation, use README as home page (56)
- Add NSGA2 and SPEA2 to QDax (33)
- Add reset_based_scoring_function for stochastic environments (31)
- Issue with batch size 1 (52)
- Add colab badges to all example notebooks (53)
- Add codecov to QDax for test coverage reports (39)
- Update singularity pipeline and fix pre-commits (45)
- Add plot_multidimensional_map_elites_grid function (47)
- Extend functionality of compute_euclidean_centroids (40)
- Repertoire addition for 1d bd (map elites and mome) (41)
- Static argnames in _sample_in_masked_pareto_front (49)
- Update ci trigger rules (48)
- Use rngkey for centroids generation
- Run workflow on pull requests
- Use RNGKey for sampling CVT centroids and for KMeans
- Fix mome emitter state update
- Add MOME to QDax (27)
- Fix steps incrementation in pointmaze (36)
- Upgrade requirements for numpy and jax (34)
- Add CMA MEGA (25) - (API change!)
- Add SMERL (DIAYN+DADS) (22)
- Add DADS (21)
- Add OMG-MEGA (24)
- Add DIAYN (20) and refactor SAC
- Add SAC algorithm (16)
- Add TD3 algorithm (14)
- Fix issue in behavior descriptor evaluation (13)