/jaxtyping

Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.

Primary LanguagePythonOtherNOASSERTION

Stargazers

No one’s star this repository yet.