jaxtyping (conda): Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.
Keywords: jax|python-typing|typing