pip 
pandas 
absl-py 
google-cloud-storage 
google-api-python-client 
oauth2client 
aenum 
tensorflow_datasets==1.3.2 
tensorflow==2.0.0
pyyaml
https://storage.googleapis.com/jax-releases/cuda100/jaxlib-0.1.37-cp36-none-linux_x86_64.whl
jax==0.1.55 
neural-tangents==0.1.4
