Add hooks for indirect learning. - #1209

Open

pfebrer wants to merge 22 commits into

metatensor:mainfrom

pfebrer:post_hooks

Commits on Sep 3, 2026

Proof of concept for hooks
pfebrer
committed
9978ea5
Add post-hooks to PET

jwa7
authored and
pfebrer
committed
dd7777c
Add gap hook
pfebrer
committed
d77a10d
fix output block with cuda values but cpu labels (#14)

cesaremalosso
and
pfebrer
committed
cde8f38
Make multipole hook work on GPU
pfebrer
committed
04748ab
Fix gap hook
pfebrer
committed
6812f48
Add tensor basis hook
pfebrer
committed
b6ce2ec
Make it serious
pfebrer
committed
be1c66e
Enable PET finetuning with hooks.
pfebrer
committed
c490b6f
Forgot to remove post_hooks from PET hypers
pfebrer
committed
83343a4
Make the dipole ansatz origin independent by subtracting the center of nuclear charge

jwa7
authored and
pfebrer
committed
d47d454
Enable post-hooks in space and soap-bpnn

jwa7
authored and
pfebrer
committed
5c67800
Make the origin a hyper of the GlobalMultipole hook, keeping the origin dependent version accessible

jwa7
authored and
pfebrer
committed
eae2b42
Fix hooks for mace

jwa7
authored and
pfebrer
committed
0d5825f
Add a note about periodic systems

jwa7
authored and
pfebrer
committed
d032bf5
Fix tensor basis hook
pfebrer
committed
996b824
Move hooks back to model hypers for MACE
pfebrer
committed
0f4cfb3
Temporary changes
pfebrer
committed
1639de9
Added PET, enable finetuning, make sure exported models work
pfebrer
committed
265011c

Commits on Sep 4, 2026

Fix restarting
pfebrer
committed
aca58fd
Register hook outputs correctly
pfebrer
committed
811d698

Commits on Sep 8, 2026

Fix MACE checkpoint updates
pfebrer
committed
ea5ba86