Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
1.05k
stars
49
forks
source link
Now supports using `Any` and `TypeVar`s as the array type. #226
Closed
patrick-kidger closed 1 week ago