HyenaLAI — Local Ancestry Inference weights
Pretrained checkpoints for HyenaLAI, a single subquadratic Hyena model that performs Local Ancestry Inference from SNPs. See the code repository for architecture, training and the zero-setup demo notebook.
Contents
One subfolder per (<classes>/<window>) configuration; each ships classifier.ckpt
(Stage-2 fine-tuned model) and config.yaml (model kwargs to rebuild it without Hydra).
examples/ holds a few per-individual HDF5s (v3) so the demo runs with no preprocessing.
| Task | Window (SNPs) | Val accuracy |
|---|---|---|
| 3 classes | 1024 | 97.03% |
| 3 classes | 4096 | 99.50% |
| 3 classes | 32768 | 99.89% |
| 5 classes | 1024 | 77.93% |
| 5 classes | 4096 | 95.28% |
3-class = EUR / AFR / EAS · 5-class = EUR / AFR / EAS / NAT / SAS (in progress).
Usage
from huggingface_hub import hf_hub_download
ckpt = hf_hub_download("gcameto/hyena-lai", "3class/4096/classifier.ckpt")
config = hf_hub_download("gcameto/hyena-lai", "3class/4096/config.yaml")
See notebooks/demo_inference.ipynb in the code repo for end-to-end whole-genome inference.
License & citation
Apache 2.0. If you use these weights, please cite the HyenaLAI paper (CIARP 2026) and thesis (UdelaR, 2026) — see the code repository for BibTeX.