gongyu-lightmatter/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.
PythonNOASSERTION
Watchers
No one’s watching this repository yet.
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.
PythonNOASSERTION
No one’s watching this repository yet.