sphinx_rtd_theme==0.5.1
Sphinx==3.4.3
matplotlib==3.1.2
flax==0.3.0
numpy==1.18.1
jsonlines==2.0.0
jax==0.2.9
jaxlib
nbsphinx
