nbsphinx
jupyter
matplotlib
sphinx-copybutton
numpy
jax
jaxlib