2. Add Apple Silicon (MPS) training support#8

Open

musicalplatypus wants to merge 5 commits into

TexasInstruments:mainfrom

musicalplatypus:pr/mps-support

Commits on Jul 16, 2026

Fix MPS (Apple Silicon) compatibility in training, evaluation and quantization
t5fkg8d44d-beep
committed
Enable NAS on MPS/CPU by resolving the device instead of hardcoding CUDA
t5fkg8d44d-beep
committed
Fix MPS float64 crash in ONNX-test and reference training scripts
t5fkg8d44d-beep
committed

Commits on Jul 20, 2026

fix: restore misplaced create_exp_dir line and add missing numpy imports
t5fkg8d44d-beep
committed

Commits on Jul 22, 2026

fix(mps): cast dtype before moving to device in utils.py evaluation loops

t5fkg8d44d-beep
and
claude
committed