
[full]
jax
jaxlib
numpyro
optax
scipy
