Gymnax

2. `spaces` now depends on `gymnasium` instead of `gym`, adding an additional dependency. Maybe later on `gym` dependency can be completely removed since it's not maintained anymore? haven't looked into that.

4. Ran everything in `python 3.10` so technically you could support that too now.
5. In `wrapper/brax.py`, `GymnaxtoBraxWrapper` has two new methods `backend` and `observation_size`, as its required to be defined under newest brax version. Used brax version `0.10.0`.

Thanks to Aidandos !

Fix import errors for `matplotlib` and `seaborn` visualization.

1. Fixed most of the typing issues under pylint --strict.

Added

- Gym, EvoJAX, Brax, DM env wrappers
- Reacher environment inspired by [Lenton et al. (2021)](https://github.com/unifyai/gym/)
- Swimmer environment inspired by [Lenton et al. (2021)](https://github.com/unifyai/gym/)
- Basic Pong environment inspired by [Kirsch (2018)](https://github.com/BlackHC/batch_pong_poc)

Fixed

- Fixed Minatar tests for jax arrays.
- Fixed reward setting in `DiscountingChain-v0`
- Fixed `Tuple` space check via enumerate.

Changed

- Refactored wrappers into separate sub-directory.

Fixed

- Fix deprecated `tree_multimap`.
- Fix device grabbing when using an `jnp.array` to set default in Acrobot env.