optax
flax
matplotlib
jax2d
wandb
safetensors
gymnax
git+https://github.com/DramaCow/jaxued.git
jaxgl
hydra-core
pygame
pygame-widgets
pygbag