ml_collections
tensorflow==2.6.1
tensorflow_datasets==4.2.0
tensorflow_addons==0.14.0
numpy==1.19.5
matplotlib==3.3.4
tqdm==4.56.0
pickle
typing
