lucidrains / equiformer-pytorch

Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein folding
MIT License
247 stars 23 forks source link

Equiformer v2 implementation #18

Open mahdip72 opened 5 months ago

mahdip72 commented 5 months ago

Hello, Thanks for your amazing repository.

Do you have any plans to complete these two? refactor into full so3 -> so2 linear layer, as proposed in eSCN paper

add equiformer v2, and start looking into equivariant protein backbone diffusion again