compositional-attention

Depends On

• tensorflow (tensorflow >= 2.5.0)
• einops (einops ~= 0.3.0)