named-tensors

There are 1 repositories under named-tensors topic.

patrick-kidger/torchtyping
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
Language:Python1.4k 14 3635