numpy
scipy
sympy
qutip
python-box
matplotlib
pandas

[full]
tensorflow
tensorflow-probability
jax
jaxlib
diffrax
equinox

[jax]
jax
jaxlib
diffrax
equinox

[tf]
tensorflow
tensorflow-probability
