flax
jax>=0.2.17
jaxlib>=0.1.65
Pillow
cma
