Constraint Dissolving Approaches for Riemannian Optimization
cdopt.linen contains predefined neural layers for Flax.
cdopt.linen