tensorflow==2.6.4
jax==0.3.13
jaxlib
numpy
matplotlib==3.5.2
gradio==3.0.15
gdown==4.4.0