scipy
numpy>=1.18.0
pillow
jaxlib
jax>=0.3.25
tensorboard>=2.3.0
parameterized
