torch==1.4.0
torchvision==0.5.0
jaxlib==0.1.46
jax==0.1.66
tensorflow==2.0.1
numba==0.47.0
matplotlib==3.2.1
pillow==7.1.2
tensorboardX==2.0
