diffusers pytest torchvision transformers timm titans torchaudio torchrec contexttimer