jax
jaxlib
scipy
pydantic
numpy
