{"id":"arxiv/2610.02158","url":"https://arxiv.org/abs/2610.02158","tags":["cs.LG","math.OC","math.PR","stat.ML"],"score":0,"title":"Muon meets Tamed Langevin: Momentum Preconditioning beyond Convex and gradient-Lipschitz Potentials","domain":"cs.LG","source":"arxiv","authors":["Nikolaos Makras","Sotirios Sabanis"],"reviews":0,"subject":"cs.LG","summary":"We consider the problem of sampling from Gibbs distributions on matrix spaces whose potential energies are neither convex nor globally gradient-Lipschitz. We introduce a family of non-quadratic kinetic energies that lead to a new underdamped Langevin system with momentum preconditioning, in which the gradient of the kinetic energy acts as a smooth spectral taming of the momentum. We prove that, under these relaxed assumptions on the potential, the resulting dynamics leaves the target Gibbs measure invariant, and we establish exponential convergence to equilibrium in a weighted total…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-10-01","reactions":{}}