Geoopt: Riemannian Optimization in PyTorch. - Max Kochurov; Rasul Karimov; Serge Kozlukov (CoRR, 2020)