Add MPS training stability fixes and experiment logs 19655a1 outlawmold commited on about 1 month ago