search

hpcaitech / FastFold

Optimizing AlphaFold Training and Inference on GPU Clusters
Apache License 2.0
557 stars 86 forks source link

support alphafold v2.3 param #128

Closed Shenggan closed 1 year ago

Shenggan commented 1 year ago

The weight import issue is introduced by _fused_triangle_multiplication: https://github.com/deepmind/alphafold/blob/9b18d6a966b9b08b2095dd77d8414a68d3d31fc9/alphafold/model/modules.py#L1384