language:en license:mit size_categories:n<1K format:imagefolder modality:image library:datasets library:mlcroissant arxiv:2502.16982 region:us optimization muon deep-learning pytorch neural-networks tutorial