Skip to content

Latest commit

 

History

History
70 lines (62 loc) · 3.63 KB

README.md

File metadata and controls

70 lines (62 loc) · 3.63 KB

xFold

A collection of folding models implemented from scratch.

Directory Structure

.
├── configs/                        # Hydra configuration directory
│   ├── config.yaml                     # Base configuration
│   ├── predict.yaml                    # Prediction configuration
│   ├── model/                          # Model-specific configuration
│   │   └── alphafold2.yaml
│   └── experiment/                     # Experiment-specific configuration
│       └── predict.yaml
├── experiments/                    # Executable experiment scripts
│   └── predict.py                      # Inference script
└── xfold/
    ├── configs/                        # Configuration files for different models
    │   ├── base.py                         # Base configuration classes
    │   ├── alphafold2.py                   # AlphaFold2 specific configurations
    │   └── esmfold.py                      # ESMFold specific configurations
    ├── model/                          # Core model implementations
    │   ├── __init__.py                     # Model registration
    │   ├── base.py                         # Base abstract model class
    │   ├── common/                         # Shared model components
    │   │   ├── aux_heads.py                    # Auxiliary model heads
    │   │   ├── misc.py                         # Miscellaneous components
    │   │   ├── structure.py                    # Structure module
    │   │   └── triangle_ops.py                 # Triangle operations
    │   ├── alphafold2/                     # AlphaFold2 specific implementation
    │   │   ├── embedders.py                    # Embedding layers
    │   │   ├── evoformer.py                    # Evoformer stack
    │   │   └── model.py                        # Main AlphaFold2 model
    │   └── esmfold/                        # ESMFold specific implementation
    │       ├── constants.py                    # ESM-specific constants
    │       ├── lm.py                           # BERT / ESM language model
    │       └── model.py                        # Main ESMFold model
    ├─── protein/                       # Protein classes and functions
    │   ├── constants.py                    # Amino-acid/atomic constants
    │   ├── sequence.py                     # MSA and sequence
    │   └── structure.py                    # Rigid frames and structure
    └─── utils/                         # Utility functions
        └── path.py                         # for handling paths

Checklist

AlphaFold2

  • Data pipeline
  • Main inference loop
    • MSA/Template featurisation
    • Evolutionary information inclusion
      • Input embedder
      • Recycling embedder
      • Template embedder
      • Extra MSA embedder
    • Evoformer stack
    • Structure module
  • Auxiliary loss heads

ESMFold

  • Main inference loop
    • ESM language model
    • Folding trunk

Resources