GitHub repository: olliethomas/torchtyping
Primary language: Python
Description: Runtime type annotations for the shape, dtype etc. of PyTorch Tensors.