mkdocs==1.4.2
mkdocs-material==9.0.3
mkdocstrings==0.19.1
mkdocstrings-python==0.8.3

jax[cpu]
