Tta pytorch implementations

hello every.
Test time augmentation (TTA) can be seen in many academic literature and kaggle solutions. I have managed to implement a ten crop augmentation when testing, where the predictions on ten crops are averaged.
Are there any implementations for similar TTA strategies?
Thank you!