flax
jax>=0.2.17
jaxlib>=0.1.65
Pillow
cma
matplotlib
pyyaml

[extra]
evosax
torchvision
pandas
procgen
brax
