patrick-kidger/torchtyping: Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
Category: typing
Language: Python
Topics: named-tensors, python-typing, pytorch, shape, tensors, typing