PyTorch Caps