More code is being arranged and will be released soon.
This the highly optimized general sparse attention/matching operator used by ASTR (accepted by CVPR2023) and [SEM] (coming soon).
More technical details can be found in ASTR_Supp.pdf.
Our code is inspired by Stratified Transformer. We also refer to its implementation, and implement our code using numba.
MIT