Manifold aware ``pytorch.optim``.
https://github.com/geoopt/geoopt.git
(1a732ef34b558f83b0de6ecb7c76e3a6743776f4)