GitHub repository: tsitsvero/VariationalStiefelOptimizer
Primary language: Python
Description: Implementation for the paper 'Momentum Stiefel Optimizer, with Applications to Suitably-Orthogonal Attention, and Optimal Transport' (ICLR 2023)