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 ⚠️
qdax/core/neuroevolution/losses/td3_loss.py	41.66%	28 Missing ⚠️
qdax/core/neuroevolution/buffers/buffer.py	37.14%	22 Missing ⚠️
qdax/environments/wrappers.py	45.94%	20 Missing ⚠️
qdax/core/containers/unstructured_repertoire.py	83.47%	19 Missing ⚠️
qdax/core/containers/mels_repertoire.py	76.47%	16 Missing ⚠️
qdax/baselines/pbt.py	76.47%	12 Missing ⚠️
qdax/tasks/brax_envs.py	76.92%	9 Missing ⚠️
tests/core_test/aurora_test.py	90.32%	9 Missing ⚠️
qdax/core/neuroevolution/networks/networks.py	82.22%	8 Missing ⚠️
... and 12 more

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

Additional details and impacted files

@@             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     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.