popgym 
wandb 
equinox 
optax 
numpy 
gymnasium 
tqdm
jaxlib
jax 