gongyu-lightmatter/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.
PythonNOASSERTION
No issues in this repository yet.
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.
PythonNOASSERTION
No issues in this repository yet.