Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
We develop NATTEN (Neighborhood Attention Extension), a Python package
with efficient C++ and CUDA kernels, which allows NA to run up to 40% faster than Swin's WSA while using up to 25% less
memory.