LICENSE
README.md
pyproject.toml
src/kfold/__init__.py
src/kfold/__main__.py
src/kfold.egg-info/PKG-INFO
src/kfold.egg-info/SOURCES.txt
src/kfold.egg-info/dependency_links.txt
src/kfold.egg-info/entry_points.txt
src/kfold.egg-info/requires.txt
src/kfold.egg-info/top_level.txt
src/kfold/cli/__init__.py
src/kfold/cli/main.py
src/kfold/cli/multigpu.py
src/kfold/cli/predict.py
src/kfold/constants/__init__.py
src/kfold/constants/atom.py
src/kfold/constants/bond.py
src/kfold/constants/ccd.py
src/kfold/constants/chain.py
src/kfold/constants/residue.py
src/kfold/constants/sequence.py
src/kfold/constants/training.py
src/kfold/data/__init__.py
src/kfold/data/layout.py
src/kfold/data/pipelines/__init__.py
src/kfold/data/pipelines/featurization.py
src/kfold/data/pipelines/prior_sampling.py
src/kfold/data/pipelines/structure_preparation.py
src/kfold/data/pipelines/tokenization.py
src/kfold/data/types/__init__.py
src/kfold/data/types/ccd.py
src/kfold/data/types/constraint.py
src/kfold/data/types/metadata.py
src/kfold/data/types/model_input.py
src/kfold/data/types/structure.py
src/kfold/data/types/tokenized.py
src/kfold/data/utils/__init__.py
src/kfold/data/utils/rdkit_utils.py
src/kfold/data/utils/io/__init__.py
src/kfold/data/utils/io/fasta.py
src/kfold/data/utils/io/structure.py
src/kfold/data/utils/simulation/__init__.py
src/kfold/data/utils/simulation/bioprior.py
src/kfold/data/utils/simulation/langevin_dynamics.py
src/kfold/data/utils/writer/__init__.py
src/kfold/data/utils/writer/gemmi_utils.py
src/kfold/data/utils/writer/writer.py
src/kfold/inference/__init__.py
src/kfold/inference/data_pipeline.py
src/kfold/inference/query.py
src/kfold/inference/runner.py
src/kfold/inference/utils.py
src/kfold/model/__init__.py
src/kfold/model/model.py
src/kfold/model/model_train.py
src/kfold/model/layers/__init__.py
src/kfold/model/layers/folding/__init__.py
src/kfold/model/layers/folding/atom_transformer.py
src/kfold/model/layers/folding/attention_pair_bias.py
src/kfold/model/layers/folding/diffusion.py
src/kfold/model/layers/folding/diffusion_transformer.py
src/kfold/model/layers/folding/embeddings.py
src/kfold/model/layers/folding/input_encoder.py
src/kfold/model/layers/folding/transition.py
src/kfold/model/layers/folding/utils.py
src/kfold/model/layers/seq_enc/__init__.py
src/kfold/model/layers/seq_enc/attention.py
src/kfold/model/layers/seq_enc/blocks.py
src/kfold/model/layers/seq_enc/moe.py
src/kfold/model/layers/seq_enc/nn.py
src/kfold/model/layers/seq_enc/rotary.py
src/kfold/model/layers/seq_enc/transformer_stack.py
src/kfold/model/layers/struct_enc/__init__.py
src/kfold/model/layers/struct_enc/backbone.py
src/kfold/model/layers/struct_enc/bb_tokenizer.py
src/kfold/model/layers/struct_enc/fa_tokenizer.py
src/kfold/model/layers/struct_enc/nn.py
src/kfold/model/layers/struct_enc/transformer_layers.py
src/kfold/model/layers/struct_enc/bb_vqvae/__init__.py
src/kfold/model/layers/struct_enc/bb_vqvae/affine_utils.py
src/kfold/model/layers/struct_enc/bb_vqvae/encoder.py
src/kfold/model/layers/struct_enc/bb_vqvae/quantizer.py
src/kfold/model/layers/struct_enc/bb_vqvae/rotary.py
src/kfold/model/layers/struct_enc/bb_vqvae/transformer.py
src/kfold/model/layers/struct_enc/fa_vqvae/__init__.py
src/kfold/model/layers/struct_enc/fa_vqvae/encoder.py
src/kfold/model/layers/struct_enc/fa_vqvae/pair_bias_attn.py
src/kfold/model/layers/struct_enc/fa_vqvae/pair_update.py
src/kfold/model/layers/struct_enc/fa_vqvae/quantizer.py
src/kfold/model/layers/struct_enc/fa_vqvae/tri_mul.py
src/kfold/model/layers/struct_enc/fa_vqvae/utils/__init__.py
src/kfold/model/layers/struct_enc/fa_vqvae/utils/coords_utils.py
src/kfold/model/layers/struct_enc/fa_vqvae/utils/data_transforms.py
src/kfold/model/layers/struct_enc/fa_vqvae/utils/residue_constants.py
src/kfold/model/layers/struct_enc/fa_vqvae/utils/rigid_utils.py
src/kfold/model/modules/__init__.py
src/kfold/model/modules/apo_module.py
src/kfold/model/modules/confidence_head.py
src/kfold/model/modules/distogram_head.py
src/kfold/model/modules/ecsi.py
src/kfold/model/modules/edm.py
src/kfold/model/modules/input_embedder.py
src/kfold/model/modules/patch_geometry.py
src/kfold/model/modules/prot_seq_encoder.py
src/kfold/model/modules/prot_struct_encoder.py
src/kfold/model/modules/rna_seq_encoder.py
src/kfold/model/modules/score_model.py
src/kfold/model/modules/tri_stack.py
src/kfold/model/primitives/__init__.py
src/kfold/model/primitives/activation.py
src/kfold/model/primitives/attention.py
src/kfold/model/primitives/dropout.py
src/kfold/model/primitives/initialize.py
src/kfold/model/primitives/linear.py
src/kfold/model/primitives/normalization.py
src/kfold/model/primitives/triangle_attention.py
src/kfold/model/primitives/triangle_multiplication.py
src/kfold/model/primitives/utils.py
src/kfold/training/__init__.py
src/kfold/training/training_module.py
src/kfold/training/dataset/__init__.py
src/kfold/training/dataset/datamodule.py
src/kfold/training/dataset/dl_sampler.py
src/kfold/training/dataset/cropper/__init__.py
src/kfold/training/dataset/cropper/alphafold.py
src/kfold/training/dataset/cropper/base.py
src/kfold/training/dataset/cropper/multi_anchor.py
src/kfold/training/dataset/cropper/utils.py
src/kfold/training/dataset/datasets/__init__.py
src/kfold/training/dataset/datasets/base.py
src/kfold/training/dataset/datasets/distillation.py
src/kfold/training/dataset/datasets/rcsb.py
src/kfold/training/dataset/datasets/train_dataset.py
src/kfold/training/dataset/datasets/val_dataset.py
src/kfold/training/dataset/sampler/__init__.py
src/kfold/training/dataset/sampler/base.py
src/kfold/training/dataset/sampler/cluster.py
src/kfold/training/dataset/sampler/uniform.py
src/kfold/training/dataset/utils/__init__.py
src/kfold/training/dataset/utils/apo_io.py
src/kfold/training/dataset/utils/apo_perturbation.py
src/kfold/training/dataset/utils/pre_crop.py
src/kfold/training/loss/__init__.py
src/kfold/training/loss/confidence.py
src/kfold/training/loss/diffusion.py
src/kfold/training/loss/distogram.py
src/kfold/training/loss/patch_geometry.py
src/kfold/training/metrics/__init__.py
src/kfold/training/metrics/structure_metrics.py
src/kfold/training/optim/__init__.py
src/kfold/training/optim/ema.py
src/kfold/training/optim/lr_scheduler.py
src/kfold/training/preprocess/__init__.py
src/kfold/training/preprocess/apo_lmdb.py
src/kfold/training/preprocess/apo_preparation.py
src/kfold/training/preprocess/atlasfold_prediction.py
src/kfold/training/preprocess/cif_factory.py
src/kfold/training/preprocess/prepared_sequences.py
src/kfold/training/preprocess/structure_cleaning.py
src/kfold/training/preprocess/structure_tokenizers.py
src/kfold/training/utils/__init__.py
src/kfold/training/utils/gradient_logging.py
src/kfold/training/utils/permutation_alignment/__init__.py
src/kfold/training/utils/permutation_alignment/align_train.py
src/kfold/training/utils/permutation_alignment/align_val.py
src/kfold/training/utils/permutation_alignment/symmetry.py
src/kfold/utils/__init__.py
src/kfold/utils/checkpointing.py
src/kfold/utils/confidence_metrics.py
src/kfold/utils/config.py
src/kfold/utils/misc.py
src/kfold/utils/mmseqs2.py
src/kfold/utils/rcsb_api.py
src/kfold/utils/registry.py
src/kfold/utils/runtime.py
src/kfold/utils/geometry/__init__.py
src/kfold/utils/geometry/random_augment.py
src/kfold/utils/geometry/rigid_align.py
src/kfold/utils/kernels/__init__.py
src/kfold/utils/kernels/cdist.py