flax
jax>=0.2.17
jaxlib>=0.1.65
Pillow
cma

[extra]
evosax
torchvision
