Repository info

jaxtyping

• Summary: a summary of jaxtyping
• Description: Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
• Stars: 1476
• Number of forks: 75