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