GitHub repository: muhd-umer/PVT-Flax
Primary language: Python
Description: Unofficial JAX/Flax implementation of Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction without Convolutions. https://arxiv.org/abs/2102.12122