Swin Transformer TensorFlow Implementation

Follow the full discussion on Reddit.
A few others and I recently implemented a TensorFlow version of Microsoft's Swin Transformer (https://arxiv.org/abs/2103.14030). It's an (almost) direct translation of the official PyTorch code (https://github.com/microsoft/Swin-Transformer) so that people can easily switch reading between the two. The TensorFlow version repo also includes code that converts PyTorch .pth weights into TensorFlow checkpoints. Using this file, you guys can either use the pretrained weights provided by Microsoft or weights of a custom model trained using PyTorch. I hope you guys find it useful!