Enable Energy and Force fitting on a GPU#72

Merged

jthorton merged 3 commits intomainfrom energy_update

Oct 8, 2024

+24-13

Commits on Aug 29, 2024

make sure energy and force predictions are on the same device to allow for GPU training
jthorton
committed

Commits on Aug 30, 2024

fix tests dtype
jthorton
committed

Commits on Oct 8, 2024

Merge branch 'main' into energy_update
jthorton
authored