/VariationalStiefelOptimizer

Implementation for the paper 'Momentum Stiefel Optimizer, with Applications to Suitably-Orthogonal Attention, and Optimal Transport' (ICLR 2023)

Primary LanguagePythonMIT LicenseMIT

Stargazers