Sync KL-Shampoo to OSS#255

runame wants to merge 1 commit intofacebookresearch:mainfacebookresearch/optimizers:mainfrom

runame:export-D89193149runame/optimizers:export-D89193149Copy head branch name to clipboard