arXiv · 2005.02819
Geoopt: Riemannian Optimization in PyTorch
Abstract
Geoopt is a research-oriented modular open-source package for Riemannian Optimization in PyTorch. The core of Geoopt is a standard Manifold interface that allows for the generic implementation of optimization algorithms. Geoopt supports basic Riemannian SGD as well as adaptive optimization algorithms. Geoopt also provides several algorithms and arithmetic methods for supported manifolds, which allow composing geometry-aware neural network layers that can be integrated with existing models.
Explore related subjects
Keep this discovery
Max Kochurov, Rasul Karimov, Serge Kozlukov. 2020-05-06. Geoopt: Riemannian Optimization in PyTorch. https://arxiv.org/abs/2005.02819
Cite the original work for its findings. Save a collection to share your selection of sources.