afrozenator / jaxtyping Goto Github PK
View Code? Open in Web Editor NEWThis project forked from patrick-kidger/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
License: Other