MurrellLab/LaProteina.jl

La-Proteina SafeTensors Weights

These are the pretrained weights for La-Proteina, converted from the original PyTorch .ckpt checkpoints to SafeTensors format.

Source

• Original repository: https://github.com/NVIDIA-Digital-Bio/la-proteina
• Paper: https://arxiv.org/abs/2507.09466
• Project page: https://research.nvidia.com/labs/genair/la-proteina/

Files

Score Networks

File	Model	Description
LD1_ucond_notri_512.safetensors	LD1	Unconditional, no triangle update (up to 512 residues)
LD2_ucond_tri_512.safetensors	LD2	Unconditional, with triangle update (up to 512 residues)
LD3_ucond_notri_800.safetensors	LD3	Unconditional, long proteins (300-800 residues)
LD4_motif_idx_aa.safetensors	LD4	Indexed motif scaffolding, all-atom (up to 256 residues)
LD5_motif_idx_tip.safetensors	LD5	Indexed motif scaffolding, tip-atom (up to 256 residues)
LD6_motif_uidx_aa.safetensors	LD6	Unindexed motif scaffolding, all-atom (up to 256 residues)
LD7_motif_uidx_tip.safetensors	LD7	Unindexed motif scaffolding, tip-atom (up to 256 residues)

Decoders (Autoencoders)

File	Model	Description
AE1_ucond_512.safetensors	AE1	Unconditional decoder (up to 512 residues)
AE2_ucond_800.safetensors	AE2	Unconditional decoder (up to 800 residues)
AE3_motif.safetensors	AE3	Motif scaffolding decoder

Conversion

These SafeTensors files were converted from the original PyTorch .ckpt checkpoints using the scripts/convert_weights_to_safetensors.py script in the La-Proteina repository. Weight tensors are stored as Float32. Only model weights are included (no optimizer state), which is why the SafeTensors files are significantly smaller than the original checkpoints.

License

These weights are licensed under the NVIDIA Open Model License. See the LICENSE file for full terms.