GitHub repository: lv416e/riemannax
Primary language: Python
Description: RiemannAX: GPU-accelerated differentiable Riemannian optimization powered by JAX