jax
jaxlib
chex
flax
pyyaml
gym>=0.26
gymnasium
