tensorflow_gpu==1.13.1
absl_py>=1.0.0
jax>=0.2.25
jaxlib>=0.1.74
jinja2>=3.0.3
matplotlib>=3.4.2
mpmath>=1.2.1
neural_tangents>=0.3.9
numpy>=1.20.2
pandas>=1.1.3
scikit_learn>=0.23.0
scipy>=1.4.1
seaborn>=0.11.2
tabulate>=0.8.9
