numpy
jax
beartype
jaxtyping
