/jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays.

Primary LanguagePythonOtherNOASSERTION

Stargazers

No one’s star this repository yet.