Consider adding JAXAtari Environments

[TheEimer]

Since the ALE is by far the slowest environment group, we could consider adding a JAXAtari environment option. They don't have all ALE envs by far (currently only 4 and none of them in the subsets...), but might be a nice extension anyway?

Would likely entail:

• Adding new env wrapper and additional lines in make_env
• Run a rough validation (e.g. comparing score distributions for the 32 config random search)
• Adapt dependencies & docs

We could also wait until at least some of the envs more important to us (BattleZone, Phoenix, DoubleDunk) are implemented.

[LabChameleon]

Yes, I agree. In particular, since EnvPool is not maintained anymore and does not work with current Jax versions.