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
Quick fix for docs not generating correctly since the last jaxtyping release #182
Closed
patrick-kidger closed 4 months ago