Rieoptax: Riemannian Optimization in JAX

10 October 2022

Pratik Jawanpuria

Papers citing "Rieoptax: Riemannian Optimization in JAX"

3 / 3 papers shown

Title
Optimal Transport Tools (OTT): A JAX Toolbox for all things Wasserstein Marco Cuturi Laetitia Meng-Papaxanthos Yingtao Tian Charlotte Bunne Geoff Davis O. Teboul OT 143 94 0 28 Jan 2022
A Practical Method for Constructing Equivariant Multilayer Perceptrons for Arbitrary Matrix Groups Marc Finzi Max Welling A. Wilson 71 185 0 19 Apr 2021
Efficient Per-Example Gradient Computations Ian Goodfellow 160 73 0 07 Oct 2015