ml-collections==0.1.0
tensorflow-gan==2.0.0
tensorflow-hub==0.12.0
tensorflow_io==0.18.0
tensorflow_datasets==4.3.0
tensorflow==2.5.0
tensorflow-probability==0.13.0
tensorflow-addons==0.13.0
absl-py==0.12.0
flax @ git+https://github.com/google/flax.git@8b2e56e93602ff9fcd62647f28b8145e0e0c10a8
jax==0.2.18
jaxlib==0.1.69
scikit-image
odl
piq
torch
torchvision
