NaViT_Implematation Patch n' Pack: NaViT, a Vision Transformer for any Aspect Ratio and Resolution Paper Link: NaViT If you find another algorithm implementation, you can go to there AnythingFromScratchV2.