Pinned
My team at Stack just open sourced our runtime shape checking library for PyTorch & Numpy. It's hugely inspired by @patrickkidger.bsky.social's Jaxtyping but fixes a few key issues we ran into (namely, compatibility with torch.jit.trace & torch.compile). Try it with `pip install dltype`