Sachin Kadyan 1b5813df76 Snapshot of the code used to run the finetuning training on the cluster of 15 AWS p4d instances.

2023-01-09 20:55:03 -05:00

..

test_data

Snapshot of the code used to run the finetuning training on the cluster of 15 AWS p4d instances.

2023-01-09 20:55:03 -05:00

__init__.py

Initial commit

2021-09-17 18:34:59 -04:00

compare_utils.py

Add low-memory attention (still needs to be incorporated)

2021-12-17 18:00:06 -05:00

config.py

Added test for make_masked_msa

2021-12-21 22:31:52 -04:00

data_utils.py

Bring tests up to speed

2021-10-19 23:07:09 -04:00

test_data_pipeline.py

Add data pipeline test

2021-12-30 22:47:34 -05:00

test_data_transforms.py

Fix bugs in data unit tests

2021-12-30 17:24:23 -05:00

test_embedders.py

Standardize code style

2021-10-16 01:17:18 -04:00

test_evoformer.py

Fixes

2022-02-03 17:06:45 -05:00

test_feats.py

Overhaul transformation code for better parity w/ AlphaFold

2022-01-03 16:47:38 -05:00

test_import_weights.py

Fix evoformer paths in some old tests

2022-04-13 23:18:57 -04:00

test_kernels.py

Add memory-efficient attention kernels

2022-04-26 02:11:18 -04:00

test_loss.py

Add type check to attention kernel

2022-04-26 16:41:03 -04:00

test_model.py

Fixes

2022-02-03 17:06:45 -05:00

test_msa.py

Fixes

2022-02-03 17:06:45 -05:00

test_outer_product_mean.py

Snapshot of the code used to run the finetuning training on the cluster of 15 AWS p4d instances.

2023-01-09 20:55:03 -05:00

test_pair_transition.py

Fix evoformer paths in some old tests

2022-04-13 23:18:57 -04:00

test_primitives.py

Improve ease of use of LMA

2022-05-12 23:21:51 -04:00

test_structure_module.py

Overhaul transformation code for better parity w/ AlphaFold

2022-01-03 16:47:38 -05:00

test_template.py

Make template code use less memory during inference

2021-10-28 18:55:57 -04:00

test_triangular_attention.py

Fixes

2022-02-03 17:06:45 -05:00

test_triangular_multiplicative_update.py

Snapshot of the code used to run the finetuning training on the cluster of 15 AWS p4d instances.

2023-01-09 20:55:03 -05:00

test_utils.py

Overhaul transformation code for better parity w/ AlphaFold

2022-01-03 16:47:38 -05:00