Add Band Gap and Magnetic Moments Prediction Heads#3

Open

AugustinLu wants to merge 5 commits intomainfrom

f-predict-gap-magmon-9256120961331400436

Commits on Apr 11, 2026

feat: add band gap and magnetic moments prediction heads

google-labs-jules[bot]
and
AugustinLu
committed
fix: allow loading older checkpoints without missing prediction head keys

google-labs-jules[bot]
and
AugustinLu
committed
style: fix flake8 E501 line too long in sevenn/checkpoint.py

google-labs-jules[bot]
and
AugustinLu
committed
fix: allow loading older checkpoints without missing prediction head keys

google-labs-jules[bot]
and
AugustinLu
committed
feat: native support for band_gap and magnetic_moments in dataloaders

google-labs-jules[bot]
and
AugustinLu
committed