Update Python extra deps:
Update Python extra deps:

- Change matplotlib version to 3.3.2 in python_extra_deps.sh (fixed compilation error with 3.2.2 and Python 3.9)
- Add rlax==0.0.3 (required by new Jax DQN & NFSP)
- Upgrade versions of chex, jax, jaxlib to newest ones.

PiperOrigin-RevId: 372884110
Change-Id: Ifa40bc6eefd6136548eccf3c508e8921cafbfe03

deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel