AlphaEnzyme checkpoints

PyTorch checkpoints for the AlphaEnzyme specificity, kinetics, and mutation tasks. The corresponding code is available in the AlphaEnzyme repository. Input data are published separately at GENTEL-Lab/AlphaEnzyme-data.

Files

Task	Checkpoint
Specificity	checkpoints/specificity/specificity_final.pt
Kinetics	checkpoints/kinetics/seed_1234/multitask_epoch18.pt
Kinetics	checkpoints/kinetics/seed_2026/multitask_epoch2.pt
Kinetics	checkpoints/kinetics/seed_3407/multitask_epoch16.pt
Mutation	checkpoints/mutation_final.pt

Download all files into the AlphaEnzyme project root:

hf download GENTEL-Lab/AlphaEnzyme --local-dir .

This recreates the checkpoints/ directory directly. To download a single checkpoint, provide its repository path to hf download.

SHA256SUMS records the SHA-256 digest of each released checkpoint. The checkpoints are provided as PyTorch .pt files; follow the repository's task-specific loading code and configuration when using them.