Stiefel manifold optimization

[calebweinreb]

Thanks for making this awesome library!

We are interested in optimization over the Stiefel manifold (simple gradient descent). Is this supported? If not, what is the minimal set of methods that have to implemented?