Skip to content

Riemax is Born!
Compare
Choose a tag to compare
@danielkelshaw danielkelshaw released this 27 Oct 15:25
· 12 commits to main since this release
v0.1.0
ba73a15

Riemax is a JAX library for Riemannian geometry, providing the ability to define induced metrics and operate on manifolds directly. This first release provides functionality for:

  1. Computing quantities of interest on the manifold.
  2. Defining geodesic dynamics on the manifold
  3. Various methods for computing the exponential / log maps, including symplectic approaches.
  4. Metropolis-Hastings sampling

and much, much more...

I have also produced documentation, available at https://riemax.danielkelshaw.com. This is currently far from exhaustive, but should give a flavour of what the library can do -- along with some simple examples.

This library is the product of my research, and so is subject to change. Please keep an eye out to see what new features are added!

Assets 2
Loading