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
20082026
-
Quick heads-up that these days I recommend https://github.com/patrick-kidger/jaxtyping over the older repository you've linked there. I learnt a lot the first time around, so the newer one is much better :)
comment by patrickkidger · 2025-11-02 · in Why don't you use dependent types? -
…gives you at least runtime checks and also a standardized form of documenting those types. https://github.com/patrick-kidger/jaxtyping
comment by dwohnitmok · 2025-10-15 · in Bringing NumPy's type-completeness score to nearly 90% -
…[3]. [1] https://github.com/beartype/beartype/ [2] https://github.com/agronholm/typeguard [3] https://github.com/patrick-kidger/jaxtyping
comment by patrickkidger · 2024-11-27 · in Python type hints may not be not for me in practice