Pinned Repositories
VariationalStiefelOptimizer
Implementation for the paper 'Momentum Stiefel Optimizer, with Applications to Suitably-Orthogonal Attention, and Optimal Transport' (ICLR 2023)
konglk1203's Repositories
konglk1203/VariationalStiefelOptimizer
Implementation for the paper 'Momentum Stiefel Optimizer, with Applications to Suitably-Orthogonal Attention, and Optimal Transport' (ICLR 2023)