[CVPR2022 - Oral] Official Jax Implementation of Learned Queries for Efficient Local Attention
Primary LanguagePythonMIT LicenseMIT