lucidrains / triangle-multiplicative-module

Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, as a standalone package for Pytorch
29Updated 3 years ago

Related projects

Alternatives and complementary repositories for triangle-multiplicative-module