Add MPS training stability fixes and experiment logs

19655a1

outlawmold commited on about 1 month ago