← All repos

patrick-kidger/jaxtyping

View on GitHub ↗ ★ 1,881 Python

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

3 mentions on Hacker News by 2 people: 0 stories, 3 comments · first 2024-11-27 · latest 2025-11-02