Type annotations and dynamic checking for a tensor's shape, dtype, names, etc. - patrick-kidger/torchtyping