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
Slightly better compatibility again -- in this case making sure we always include an annotation. #222
Closed
patrick-kidger closed 2 weeks ago