https://www.freshports.org/devel/py-jaxtyping/
FreshPorts -- devel/py-jaxtyping: Type annotations & runtime checking for shape and dtype of JAX...
jaxtyping is a library providing type annotations and runtime type-checking for: * shape and dtype of JAX arrays * PyTrees