Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/apple/ml-simplefold
/ functions
Functions
396 in github.com/apple/ml-simplefold
⨍
Functions
396
◇
Types & classes
102
Method
_basic_init
(module)
src/simplefold/model/torch/layers.py:250
Method
_basic_init
(module)
src/simplefold/model/torch/blocks.py:46
Function
aatype_to_str_sequence
(aatype)
src/simplefold/utils/residue_constants.py:1312
Function
bootstrap_ci
Compute 95% bootstrap confidence intervals for the mean of 'series'.
src/simplefold/evaluation/analyze_folding.py:136
Method
clear
(self)
src/simplefold/evaluation/analyze_two_state.py:150
Function
compute_all_coords_mask
(structure)
src/simplefold/boltz_data_pipeline/feature/symmetry.py:388
Method
compute_alpha_t
Compute the data coefficient along the path
src/simplefold/model/flow.py:91
Function
compute_covalent_ligands
Compute the covalent ligands from a list of connections. Parameters ---------- connections: List[gemmi.Connection] The connection
src/simplefold/utils/mmcif_utils.py:276
Method
compute_d_alpha_alpha_ratio_t
Compute the ratio between d_alpha and alpha
src/simplefold/model/flow.py:99
Method
compute_sigma_t
Compute the noise coefficient along the path
src/simplefold/model/flow.py:95
Function
compute_symmetry_idx_dictionary
(data)
src/simplefold/boltz_data_pipeline/feature/symmetry.py:78
Method
compute_velocity_from_score
(self, s_t, y_t, t)
src/simplefold/model/flow.py:74
Method
configure_optimizers
(self)
src/simplefold/model/simplefold.py:764
Method
crop
Crop the data to a maximum number of tokens. Parameters ---------- data : Tokenized The tokenized data. m
src/simplefold/boltz_data_pipeline/crop/slice.py:64
Method
crop
Crop the data to a maximum number of tokens. Parameters ---------- data : Tokenized The tokenized data. m
src/simplefold/boltz_data_pipeline/crop/boltz.py:161
Function
default
(v, d)
src/simplefold/utils/boltz_utils.py:36
Function
default
returns x if it exists, otherwise y
src/simplefold/model/mlx/layers.py:133
Method
dump
Dump the object to an NPZ file. Parameters ---------- path : Path The path to the file.
src/simplefold/boltz_data_pipeline/types.py:42
Function
enforce_tags
Prompts user to input tags from command line if no tags are provided in config. :param cfg: A DictConfig composed by Hydra. :param save_to_fi
src/simplefold/utils/utils.py:89
Method
filter
Filter a chains based on their length. Parameters ---------- structure : Structure The structure to filter chains
src/simplefold/boltz_data_pipeline/filter/static/polymer.py:41
Method
filter
Filter proteins with all unknown residues. Parameters ---------- structure : Structure The structure to filter ch
src/simplefold/boltz_data_pipeline/filter/static/polymer.py:75
Method
filter
Filter protein if consecutive CA atoms above a threshold. Parameters ---------- structure : Structure The structu
src/simplefold/boltz_data_pipeline/filter/static/polymer.py:125
Method
filter
Filter out clashing chains. Parameters ---------- structure : Structure The structure to filter chains from.
src/simplefold/boltz_data_pipeline/filter/static/polymer.py:209
Method
filter
Filter chains in a structure. Parameters ---------- structure : Structure The structure to filter chains from.
src/simplefold/boltz_data_pipeline/filter/static/filter.py:20
Method
filter
Filter excluded ligands. Parameters ---------- structure : Structure The structure to filter chains from.
src/simplefold/boltz_data_pipeline/filter/static/ligand.py:153
Method
filter
Filter a data record. Parameters ---------- record : Record The object to consider filtering in / out. R
src/simplefold/boltz_data_pipeline/filter/dynamic/subset.py:33
Method
filter
Filter a data record. Parameters ---------- record : Record The object to consider filtering in / out. R
src/simplefold/boltz_data_pipeline/filter/dynamic/filter.py:18
Method
filter
Filter complexes based on their resolution. Parameters ---------- record : Record The record to filter.
src/simplefold/boltz_data_pipeline/filter/dynamic/resolution.py:27
Method
filter
Filter structures based on their resolution. Parameters ---------- record : Record The record to filter.
src/simplefold/boltz_data_pipeline/filter/dynamic/max_residues.py:30
Method
filter
Filter a record based on its date. Parameters ---------- record : Record The record to filter. Returns
src/simplefold/boltz_data_pipeline/filter/dynamic/date.py:51
Method
forward
(self, x, **kwargs)
src/simplefold/model/torch/layers.py:51
Method
forward
(self, x, **kwargs)
src/simplefold/model/torch/layers.py:88
Method
forward
(self, x)
src/simplefold/model/torch/layers.py:141
Method
forward
(self, t)
src/simplefold/model/torch/layers.py:194
Method
forward
(self, cond, train, force_drop_ids=None)
src/simplefold/model/torch/layers.py:226
Method
forward
(self, x, c)
src/simplefold/model/torch/layers.py:264
Method
forward
(self, x)
src/simplefold/model/torch/layers.py:295
Method
forward
( self, latents, c, **kwargs, )
src/simplefold/model/torch/blocks.py:58
Method
forward
( self, latents, **kwargs, )
src/simplefold/model/torch/blocks.py:105
Method
forward
(self, latents, c, **kwargs)
src/simplefold/model/torch/blocks.py:121
Method
forward
(self, noised_pos, t, feats, self_cond=None)
src/simplefold/model/torch/architecture.py:172
Method
forward
(self, pos)
src/simplefold/model/torch/pos_embed.py:20
Method
forward
Get the positional encoding for each coordinate. Args: pos: (*, in_dim) Returns: out:
src/simplefold/model/torch/pos_embed.py:94
Method
forward
xq: [B, H, N, D] xk: [B, H, N, D] pos: [B, N, in_dim]
src/simplefold/model/torch/pos_embed.py:167
Method
forward
( self, latent, feats, )
src/simplefold/model/torch/confidence_module.py:55
Method
from_pretrained_folding_model
(self)
src/simplefold/wrapper.py:78
Method
from_pretrained_plddt_model
(self)
src/simplefold/wrapper.py:117
Method
get_atoms
(self)
src/simplefold/boltz_data_pipeline/write/mmcif.py:128
Function
get_current_idx_list
(data)
src/simplefold/boltz_data_pipeline/feature/symmetry.py:93
Function
get_symmetries
Create a dictionary for the ligand symmetries. Parameters ---------- path : str The path to the ligand symmetries. Returns
src/simplefold/boltz_data_pipeline/feature/symmetry.py:43
Function
get_unk_token
Get the unknown token for a given entity type. Parameters ---------- dtype : gemmi.EntityType The entity type. Returns -
src/simplefold/utils/mmcif_utils.py:210
Method
load
Load the object from an NPZ file. Parameters ---------- path : Path The path to the file. Returns
src/simplefold/boltz_data_pipeline/types.py:26
Method
load
Load the object from a JSON file. Parameters ---------- path : Path The path to the file. Returns
src/simplefold/boltz_data_pipeline/types.py:58
Method
load
Load a structure from an NPZ file. Parameters ---------- path : Path The path to the file. Returns
src/simplefold/boltz_data_pipeline/types.py:160
Function
log
(t, eps=1e-20)
src/simplefold/utils/boltz_utils.py:40
Function
main
()
src/simplefold/cli.py:14
Function
make_atom14_dists_bounds
compute upper and lower bounds for bonds to assess violations.
src/simplefold/utils/residue_constants.py:1224
Function
map_plddt_torch_to_mlx
(key, value)
src/simplefold/utils/mlx_utils.py:79
Function
map_torch_to_mlx
(key, value)
src/simplefold/utils/mlx_utils.py:49
Function
minimum_lddt_symmetry_coords
( coords: torch.Tensor, feats: dict, index_batch: int, **args_rmsd, )
src/simplefold/boltz_data_pipeline/feature/symmetry.py:247
Function
minimum_symmetry_coords
( coords: torch.Tensor, feats: dict, index_batch: int, **args_rmsd, )
src/simplefold/boltz_data_pipeline/feature/symmetry.py:113
Method
on_before_optimizer_step
(self, optimizer: Optimizer)
src/simplefold/model/simplefold.py:685
Method
on_before_zero_grad
(self, optimizer: Optimizer)
src/simplefold/model/simplefold.py:710
Method
on_load_checkpoint
(self, checkpoint)
src/simplefold/model/simplefold.py:748
Method
on_predict_epoch_end
Print the number of failed examples.
src/simplefold/boltz_data_pipeline/write/writer.py:230
Method
on_predict_epoch_end
(self)
src/simplefold/model/simplefold.py:610
Method
on_save_checkpoint
(self, checkpoint)
src/simplefold/model/simplefold.py:727
Method
on_train_batch_end
(self, outputs, batch, batch_idx)
src/simplefold/model/simplefold.py:674
Method
on_train_start
(self)
src/simplefold/model/simplefold.py:507
Function
parse_a3m
Process an A3M file. Parameters ---------- path : Path The path to the a3m(.gz) file. taxonomy : Redis The taxonomy d
src/simplefold/boltz_data_pipeline/parse/a3m.py:112
Function
parse_csv
Process an A3M file. Parameters ---------- path : Path The path to the a3m(.gz) file. max_seqs : int, optional The ma
src/simplefold/boltz_data_pipeline/parse/csv.py:19
Function
parse_fasta
Parse a fasta file. The name of the fasta file is used as the name of this job. We rely on the fasta record id to determine the entity type.
src/simplefold/boltz_data_pipeline/parse/fasta.py:19
Function
parse_yaml
Parse a Boltz input yaml / json. The input file should be a yaml file with the following format: sequences: - protein: i
src/simplefold/boltz_data_pipeline/parse/yaml.py:18
Function
pick_interface_token
Pick a random token from an interface. Parameters ---------- tokens : np.ndarray The token data. interface : int The
src/simplefold/boltz_data_pipeline/crop/boltz.py:76
Method
predict_contacts
(self, tokens)
src/simplefold/model/mlx/esm_network.py:159
Method
predict_step
(self, batch, batch_idx)
src/simplefold/model/simplefold.py:513
Method
process_input
(self, aa_seq)
src/simplefold/wrapper.py:293
Method
register
(self, name, tensor)
src/simplefold/model/simplefold.py:140
Function
renumber_pdb
(molseq, X_path, X_renum, start=1)
src/simplefold/evaluation/analyze_two_state.py:173
Method
run_inference
(self, batch, model, plddt_model, device)
src/simplefold/wrapper.py:321
Method
sample
(self, model_fn, flow, noise, batch)
src/simplefold/model/torch/sampler.py:83
Method
save_result
(self, structure, record, results, out_name)
src/simplefold/wrapper.py:369
Function
select_subset_from_mask
(mask, p)
src/simplefold/boltz_data_pipeline/feature/featurizer.py:28
Function
sequence_to_onehot
Maps the given sequence into a one-hot encoded matrix. Args: sequence: An amino acid sequence. mapping: A dictionary mapping amino ac
src/simplefold/utils/residue_constants.py:890
Method
setup
Lightning hook that is called at the beginning of fit (train + validate), validate, test, or predict. This is a good hook when you ne
src/simplefold/model/simplefold.py:622
Method
setup
Run the setup for the DataModule. Parameters ---------- stage : str, optional The stage, one of 'fit', 'validate'
src/simplefold/datasets/test_datamodule.py:174
Method
setup
Run the setup for the DataModule. Parameters ---------- stage : str, optional The stage, one of 'fit', 'validate'
src/simplefold/datasets/train_datamodule.py:354
Method
step
(self, epoch=None)
src/simplefold/utils/lr_scheduler.py:48
Function
task_wrapper
Optional decorator that controls the failure behavior when executing the task function. This wrapper can be used to: - make sure loggers
src/simplefold/utils/utils.py:160
Function
tm_diversity
Calculate diversity among pairs of modeled protein structures for each score. Returns: dict of scores('rmsd': float, 'tm': float, 'gd
src/simplefold/evaluation/two_state_utils/tm_ens.py:84
Function
tm_ensemble_at_N
(samples, gt1, gt2, N=5)
src/simplefold/evaluation/two_state_utils/tm_ens.py:64
Method
tokenize
Tokenize the input data. Parameters ---------- data : Input The input data. Returns -------
src/simplefold/boltz_data_pipeline/tokenize/boltz_protein.py:42
Method
training_step
(self, batch, batch_idx)
src/simplefold/model/simplefold.py:496
Method
val_dataloader
Get the validation dataloader. Returns ------- DataLoader The validation dataloader.
src/simplefold/datasets/train_datamodule.py:392
Method
validation_step
(self, batch, batch_idx)
src/simplefold/model/simplefold.py:503
Method
write_on_batch_end
Write the predictions to disk.
src/simplefold/boltz_data_pipeline/write/writer.py:59
← previous
301–396 of 396, ranked by callers