Fix AdEMAMix scheduler guard and add state_dict round-trip test#1861

matthewdouglas merged 2 commits intomainbitsandbytes-foundation/bitsandbytes:mainfrom

fix/issue-1382bitsandbytes-foundation/bitsandbytes:fix/issue-1382Copy head branch name to clipboard