numpy
scipy
jax
arrayer<0.2
ray
beartype
jaxtyping
