deepfold.utils.import_utils.import_jax_weights_

deepfold.utils.import_utils.import_jax_weights_(model: AlphaFold, npz_path: str, is_multimer: bool = False, enable_ptm: bool = False, enable_templates: bool = False, fuse_projection_weights: bool = False) → None

Import AlphaFold JAX parameters.

Parameters:

• model – AlphaFold model.

• npz_path – Path to an NPZ file.

• is_multimer – Whether multimer model or not.

• enable_ptm – Enable predicted aligned error related modules.

• enable_templates – Enable template related modules.

• fuse_projection_weights – Whether triangular multiplicative layers are fused or not.