-
Notifications
You must be signed in to change notification settings - Fork 0
Make bulk inference easier & output internal embeddings #1
base: main
Are you sure you want to change the base?
Commits on Oct 6, 2021
-
Changes to make bulk inference easier:
- load in features if already computed - allow computing features for multiple fastas in parallel - allow disabling Amber relaxation, b/c our sequences have X's
Prasanna Vasudevan authored and Prasanna Vasudevan committedOct 6, 2021 Configuration menu - View commit details
-
Copy full SHA for 1bee52e - Browse repository at this point
Copy the full SHA 1bee52eView commit details -
Output internal embeddings as part of result_model_*.pkl
Prasanna Vasudevan authored and Prasanna Vasudevan committedOct 6, 2021 Configuration menu - View commit details
-
Copy full SHA for 9c00671 - Browse repository at this point
Copy the full SHA 9c00671View commit details
Commits on Oct 7, 2021
-
Prasanna Vasudevan committed
Oct 7, 2021 Configuration menu - View commit details
-
Copy full SHA for c98b294 - Browse repository at this point
Copy the full SHA c98b294View commit details -
Noticed that re-using compiled Jax model is 2-10x faster! So now,
sorting fastas by seq len and padding to round up to nearest 50, so many consecutive sequences can be run by the same compiled Jax model. Note that different models (1 through 5) have different sizes (e.g. extra msa size), so for now you must choose exactly one model in run_docker.py. In the future, may want to refactor run_alphafold.py so the outer loop is over models instead of inputs.
Prasanna Vasudevan committedOct 7, 2021 Configuration menu - View commit details
-
Copy full SHA for 74beba9 - Browse repository at this point
Copy the full SHA 74beba9View commit details
Commits on Oct 9, 2021
-
Run inference on multiple GPUs in parallel.
On a related note, tried running multiple inputs on a single GPU in separate child processes, but that was actually slower than doing it serially.
Prasanna Vasudevan committedOct 9, 2021 Configuration menu - View commit details
-
Copy full SHA for 0c23382 - Browse repository at this point
Copy the full SHA 0c23382View commit details
Commits on Oct 11, 2021
-
- split up feature generation vs model running - run one model at a time, thru all inputs (for Jax compile speedup), and then rank models at the end. - don't run a model if output already exists
Prasanna Vasudevan authored and Prasanna Vasudevan committedOct 11, 2021 Configuration menu - View commit details
-
Copy full SHA for cdaf2fa - Browse repository at this point
Copy the full SHA cdaf2faView commit details
Commits on Jan 27, 2022
-
Add line in run_docker to remind users to mount repo dir to avoid Doc…
…ker re-build on code changes
Prasanna Vasudevan committedJan 27, 2022 Configuration menu - View commit details
-
Copy full SHA for 803a11d - Browse repository at this point
Copy the full SHA 803a11dView commit details
Commits on Jan 28, 2022
-
Fix jax and jaxlib versions getting out of sync
Prasanna Vasudevan committedJan 28, 2022 Configuration menu - View commit details
-
Copy full SHA for 3b882f7 - Browse repository at this point
Copy the full SHA 3b882f7View commit details
Commits on Jan 29, 2022
-
Don't fail if can't find a PDB .cif template file
(workaround for now, in the interest of time)
Prasanna Vasudevan committedJan 29, 2022 Configuration menu - View commit details
-
Copy full SHA for 8edaafd - Browse repository at this point
Copy the full SHA 8edaafdView commit details
Commits on Jan 30, 2022
-
Don't run a model if output already exists
Prasanna Vasudevan committedJan 30, 2022 Configuration menu - View commit details
-
Copy full SHA for f459ad4 - Browse repository at this point
Copy the full SHA f459ad4View commit details -
Minor change, forgot to commit earlier
Prasanna Vasudevan committedJan 30, 2022 Configuration menu - View commit details
-
Copy full SHA for 0d737df - Browse repository at this point
Copy the full SHA 0d737dfView commit details
Commits on Feb 5, 2022
-
After refactor of run_alphafold, forgot to re-enable ranking of model…
…s at the end. Do that now.
Prasanna Vasudevan committedFeb 5, 2022 Configuration menu - View commit details
-
Copy full SHA for ab59843 - Browse repository at this point
Copy the full SHA ab59843View commit details
Commits on Feb 16, 2022
-
Average embeddings across amino acids and output to a separate, conve…
…nient file
Prasanna Vasudevan committedFeb 16, 2022 Configuration menu - View commit details
-
Copy full SHA for cf48f3e - Browse repository at this point
Copy the full SHA cf48f3eView commit details -
Prasanna Vasudevan committed
Feb 16, 2022 Configuration menu - View commit details
-
Copy full SHA for 69a289a - Browse repository at this point
Copy the full SHA 69a289aView commit details
Commits on Feb 21, 2022
-
AlphaFold runner: skip model if result .pkl file OR just the small em…
…beddings .pkl is outputted. Also, when ranking models, grab plddt from the small embeddings .pkl file, not the huge result .pkl file.
Prasanna Vasudevan committedFeb 21, 2022 Configuration menu - View commit details
-
Copy full SHA for fc79d58 - Browse repository at this point
Copy the full SHA fc79d58View commit details
Commits on Feb 22, 2022
-
Catch exception in run_model instead of exiting
Prasanna Vasudevan committedFeb 22, 2022 Configuration menu - View commit details
-
Copy full SHA for 74d245e - Browse repository at this point
Copy the full SHA 74d245eView commit details -
Fix outputting of embeddings / pLDDT
Prasanna Vasudevan committedFeb 22, 2022 Configuration menu - View commit details
-
Copy full SHA for 3241dec - Browse repository at this point
Copy the full SHA 3241decView commit details -
Output full, per-token embeddings in addition to average-token embedd…
…ings
Prasanna Vasudevan committedFeb 22, 2022 Configuration menu - View commit details
-
Copy full SHA for 9e5dd19 - Browse repository at this point
Copy the full SHA 9e5dd19View commit details
Commits on Mar 31, 2022
-
Catch exception in generating features, rather than failing.
Prasanna Vasudevan committedMar 31, 2022 Configuration menu - View commit details
-
Copy full SHA for 48f9776 - Browse repository at this point
Copy the full SHA 48f9776View commit details