.. jax_rb documentation master file, created by
   sphinx-quickstart on Mon Apr  1 23:16:45 2024.
   You can adapt this file completely to your liking, but it should at least
   contain the root `toctree` directive.

JAX RB: Riemannian Brownian motion on manifolds
===============================================
JAX RB is a library for Riemannian Brownian motion on constrained manifolds embedded in a vector space, or on quotient of such manifolds by a compact group. The theory is developed in [NgSo24]_. Please refer to the `github repository <https://github.com/dnguyend/jax-rb>`_ for installation instruction.

.. math::
.. toctree::
   :maxdepth: 2
   :caption: Manifolds
	     
   manifolds

.. toctree::
   :maxdepth: 2
   :caption: Simulation
	     
   simulation


.. toctree::
   :maxdepth: 2
   :caption: Utils
	     
   utils
   


Indices and tables
==================

* :ref:`genindex`
* :ref:`modindex`
* :ref:`search`


.. toctree::
   :maxdepth: 2
   :caption: Other Resources:

   Wikipedia <https://en.wikipedia.org/wiki/Stochastic_analysis_on_manifolds>
   GitHub <https://github.com/dnguyend/jax-rb>

   animation	     

.. rubric:: References   
.. [NgSo24] Nguyen, D.; Sommer, S...: Second-order differential operators, stochastic differential equations and Brownian motions on embedded manifolds `arXiv:2406.02879 <https://arxiv.org/abs/2406.02879>`_.