jax
einops
equinox
diffrax
matplotlib
jaxtyping
ml-dtypes
numpy
plum-dispatch
wadler_lindig
tqdm
scipy

[dev]
pytest
