tensorflow==2.7.0
jax==0.2.25
pandas==1.3.4
neural-tangents==0.3.9
jaxlib==0.1.73
mpmath==1.2.1
scikit-learn==1.0.1
matplotlib
tabulate
seaborn
Jinja2
yapf
toml
isort
