Skip to content

Commit

Permalink
add arxiv link
Browse files Browse the repository at this point in the history
  • Loading branch information
dnguyend authored Jun 6, 2024
1 parent 6b27086 commit 810e12c
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
# JAX_RB: Riemannian Brownian motion
This package provides a framework to implement Brownian motion on Riemannian manifolds. Check out the [documentation page](https://dnguyend.github.io/jax-rb/index.html).
This package provides a framework to implement Brownian motion on Riemannian manifolds. The theory is developed in [this paper](https://arxiv.org/abs/2406.02879).
Check out the [documentation page](https://dnguyend.github.io/jax-rb/index.html).
## Installation
Requirement: [JAX](https://jax.readthedocs.io/en/latest/installation.html) (pip install jax). If we have access to GPU then install jax cuda following JAX's installation note (for example pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html).

Expand Down

0 comments on commit 810e12c

Please sign in to comment.