├── .gitattributes ├── .gitignore ├── LICENSE.txt ├── MODEL_CARD.md ├── README.md ├── assets ├── fig1.png └── image.png ├── model_weights ├── trained_on_afdb.ckpt └── trained_on_pdb.ckpt ├── multiflow_config.yaml ├── proteinblobs ├── blobs.py ├── constants.py ├── dataset.py ├── designability_utils.py ├── logger.py ├── multiflow │ ├── LICENSE.txt │ ├── cross_attention.py │ ├── data │ │ ├── all_atom.py │ │ ├── interpolant.py │ │ ├── protein.py │ │ ├── residue_constants.py │ │ ├── so3_utils.py │ │ └── utils.py │ ├── datasets.py │ ├── edge_feature_net.py │ ├── flow_model.py │ ├── ipa_pytorch.py │ ├── node_feature_net.py │ ├── rigid_utils.py │ └── utils.py ├── multiflow_wrapper.py ├── parsing.py ├── utils.py ├── visualize.py └── wrapper.py ├── run_multiflow.py ├── sample.py ├── scripts ├── ca_only_helicity.py ├── create_multiflowdb.py ├── evaluate_alignment.py ├── evaluate_designability.py ├── evaluate_designability_codesign.py ├── preprocess_clusters.py └── unpack_geniepdb.py └── train.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /MODEL_CARD.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/MODEL_CARD.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/README.md -------------------------------------------------------------------------------- /assets/fig1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/assets/fig1.png -------------------------------------------------------------------------------- /assets/image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/assets/image.png -------------------------------------------------------------------------------- /model_weights/trained_on_afdb.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/model_weights/trained_on_afdb.ckpt -------------------------------------------------------------------------------- /model_weights/trained_on_pdb.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/model_weights/trained_on_pdb.ckpt -------------------------------------------------------------------------------- /multiflow_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/multiflow_config.yaml -------------------------------------------------------------------------------- /proteinblobs/blobs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/blobs.py -------------------------------------------------------------------------------- /proteinblobs/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/constants.py -------------------------------------------------------------------------------- /proteinblobs/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/dataset.py -------------------------------------------------------------------------------- /proteinblobs/designability_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/designability_utils.py -------------------------------------------------------------------------------- /proteinblobs/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/logger.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/LICENSE.txt -------------------------------------------------------------------------------- /proteinblobs/multiflow/cross_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/cross_attention.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/data/all_atom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/data/all_atom.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/data/interpolant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/data/interpolant.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/data/protein.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/data/protein.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/data/residue_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/data/residue_constants.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/data/so3_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/data/so3_utils.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/data/utils.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/datasets.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/edge_feature_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/edge_feature_net.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/flow_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/flow_model.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/ipa_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/ipa_pytorch.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/node_feature_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/node_feature_net.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/rigid_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/rigid_utils.py -------------------------------------------------------------------------------- /proteinblobs/multiflow/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow/utils.py -------------------------------------------------------------------------------- /proteinblobs/multiflow_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/multiflow_wrapper.py -------------------------------------------------------------------------------- /proteinblobs/parsing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/parsing.py -------------------------------------------------------------------------------- /proteinblobs/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/utils.py -------------------------------------------------------------------------------- /proteinblobs/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/visualize.py -------------------------------------------------------------------------------- /proteinblobs/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/proteinblobs/wrapper.py -------------------------------------------------------------------------------- /run_multiflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/run_multiflow.py -------------------------------------------------------------------------------- /sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/sample.py -------------------------------------------------------------------------------- /scripts/ca_only_helicity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/scripts/ca_only_helicity.py -------------------------------------------------------------------------------- /scripts/create_multiflowdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/scripts/create_multiflowdb.py -------------------------------------------------------------------------------- /scripts/evaluate_alignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/scripts/evaluate_alignment.py -------------------------------------------------------------------------------- /scripts/evaluate_designability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/scripts/evaluate_designability.py -------------------------------------------------------------------------------- /scripts/evaluate_designability_codesign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/scripts/evaluate_designability_codesign.py -------------------------------------------------------------------------------- /scripts/preprocess_clusters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/scripts/preprocess_clusters.py -------------------------------------------------------------------------------- /scripts/unpack_geniepdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/scripts/unpack_geniepdb.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/protcomposer/HEAD/train.py --------------------------------------------------------------------------------