compositional-attention-pytorch

Depends On

• einops (einops>=0.4)
• einops-exts (einops-exts)
• torch (torch>=1.6)