Force float32 precision on CPU/MPS instead of bf16-mixed - #258

Open

fnachon wants to merge 9 commits into

HannesStark:mainfrom

fnachon:fix/force-fp32-precision-cpu-mps

Commits on Jan 10, 2026

Minor adaptations for Mac GPU MPS compatibility
fnachon
committed
3173659
Merge branch 'HannesStark:main' into main
fnachon
authored
f84684f

Commits on Mar 31, 2026

Merge remote-tracking branch 'upstream/main'
fnachon
committed
61a248d
Fix MPS compatibility for new upstream code

fnachon
and
claude
committed
0327685
Fix KeyError 'name' by reloading molecules fresh from moldir in workers

fnachon
and
claude
committed
4bf9161
Fix MPS pin_memory warning and persistent_workers hint in all DataLoaders

fnachon
and
claude
committed
aeba138
Update README.md
fnachon
committed
698d095

Commits on Jun 30, 2026

Merge remote-tracking branch 'upstream/main'
fnachon
committed
592317f

Commits on Jul 10, 2026

Force float32 precision on CPU/MPS instead of bf16-mixed

fnachon
and
claude
committed
bb23918