fix: fix all notebooks to run with latest develop

[manon-but-yes]

Fix all the notebooks so they run with the latest code from develop branch. All the notebooks have been run with latest develop as a check.

This PR introduces:

• fix all dependencies in notebooks.
• propagate all code changes of develop to notebooks.
• update Aurora and PGA-Aurora notebooks to match the implementation merged on develop.

[codecov-commenter]

Codecov Report

Attention: 238 lines in your changes are missing coverage. Please review.

Comparison is base (f6a0b71) 92.31% compared to head (5f74f60) 90.98%. Report is 29 commits behind head on develop.

Files	Patch %	Lines
qdax/environments/humanoidtrap.py	18.30%	58 Missing :warning:
qdax/core/neuroevolution/losses/td3_loss.py	41.66%	28 Missing :warning:
qdax/core/neuroevolution/buffers/buffer.py	37.14%	22 Missing :warning:
qdax/environments/wrappers.py	45.94%	20 Missing :warning:
qdax/core/containers/unstructured_repertoire.py	83.47%	19 Missing :warning:
qdax/core/containers/mels_repertoire.py	76.47%	16 Missing :warning:
qdax/baselines/pbt.py	76.47%	12 Missing :warning:
qdax/tasks/brax_envs.py	76.92%	9 Missing :warning:
tests/core_test/aurora_test.py	90.32%	9 Missing :warning:
qdax/core/neuroevolution/networks/networks.py	82.22%	8 Missing :warning:
... and 12 more

:exclamation: Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## develop #169 +/- ## =========================================== - Coverage 92.31% 90.98% -1.34% =========================================== Files 103 125 +22 Lines 5816 7653 +1837 =========================================== + Hits 5369 6963 +1594 - Misses 447 690 +243 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.