# Core
jax
jaxlib
equinox
optax
numpyro