stable-gymnax
dm_pix
jaxtyping

[all]
pygame-ce
memax[equinox]
optax
equinox
distreqx
wandb[media]
beartype
jaxtyping
imageio
pytest

[baselines]
memax[equinox]
optax
equinox
distreqx
wandb[media]
beartype
jaxtyping
imageio

[human]
pygame-ce

[tests]
pytest
