Merge branch 'main' of https://github.com/hpcaitech/FastFold

b14e47f4 · zhuwenwen · 490cb6f5 · 05681304 · b14e47f4 · b14e47f4
Commit b14e47f4 authored Apr 26, 2023 by zhuwenwen
20 changed files
--- a/fastfold/model/fastnn/kernel/jit/__init__.py
+++ b/fastfold/model/fastnn/kernel/jit/__init__.py
--- a/fastfold/model/fastnn/kernel/jit/fused_ops.py
+++ b/fastfold/model/fastnn/kernel/jit/fused_ops.py
--- a/fastfold/model/fastnn/kernel/jit/options.py
+++ b/fastfold/model/fastnn/kernel/jit/options.py
--- a/fastfold/model/fastnn/kernel/layer_norm.py
+++ b/fastfold/model/fastnn/kernel/layer_norm.py
--- a/fastfold/model/fastnn/kernel/softmax.py
+++ b/fastfold/model/fastnn/kernel/softmax.py
--- a/fastfold/model/fastnn/kernel/triton/__init__.py
+++ b/fastfold/model/fastnn/kernel/triton/__init__.py
--- a/fastfold/model/fastnn/kernel/triton/attention_core.py
+++ b/fastfold/model/fastnn/kernel/triton/attention_core.py
--- a/fastfold/model/fastnn/kernel/triton/layer_norm.py
+++ b/fastfold/model/fastnn/kernel/triton/layer_norm.py
--- a/fastfold/model/fastnn/kernel/triton/softmax.py
+++ b/fastfold/model/fastnn/kernel/triton/softmax.py
--- a/fastfold/model/fastnn/msa.py
+++ b/fastfold/model/fastnn/msa.py
--- a/fastfold/model/fastnn/ops.py
+++ b/fastfold/model/fastnn/ops.py
--- a/fastfold/model/fastnn/template.py
+++ b/fastfold/model/fastnn/template.py
--- a/fastfold/model/fastnn/triangle.py
+++ b/fastfold/model/fastnn/triangle.py
--- a/fastfold/model/hub/__init__.py
+++ b/fastfold/model/hub/__init__.py
--- a/fastfold/model/hub/alphafold.py
+++ b/fastfold/model/hub/alphafold.py
--- a/fastfold/model/hub/loss.py
+++ b/fastfold/model/hub/loss.py
--- a/fastfold/model/hub/lr_scheduler.py
+++ b/fastfold/model/hub/lr_scheduler.py
--- a/fastfold/model/nn/__init__.py
+++ b/fastfold/model/nn/__init__.py
--- a/fastfold/model/nn/dropout.py
+++ b/fastfold/model/nn/dropout.py
--- a/fastfold/model/nn/embedders.py
+++ b/fastfold/model/nn/embedders.py