lucidrains / compositional-attention-pytorch

Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process with disentangled search and retrieval head aggregation, in Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

lucidrains/compositional-attention-pytorch Stargazers