Index

_

_lie_algebra_proj() (in module jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant)

_mat_apply() (in module jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant)

A

AffineLeftInvariant (class in jax_rb.manifolds.affine_left_invariant)

approx_nearest() (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method)

C

complement_basis_for_vector() (in module jax_rb.utils.utils)

D

DiagHypersurface (class in jax_rb.manifolds.diag_hypersurface)

E

esqrtm() (in module jax_rb.utils.utils)

G

g_metric() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method) gamma() (jax_rb.manifolds.global_manifold.GlobalManifold method) gamma_ambient() (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method) generate_symmetric_tensor() (in module jax_rb.utils.utils) geodesic_move() (in module jax_rb.simulation.global_manifold_integrator) (in module jax_rb.simulation.matrix_group_integrator) geodesic_move_dim_g() (in module jax_rb.simulation.matrix_group_integrator) geodesic_move_dim_g_normalized() (in module jax_rb.simulation.matrix_group_integrator)

geodesic_move_exact() (in module jax_rb.simulation.global_manifold_integrator) geodesic_move_exact_normalized() (in module jax_rb.simulation.global_manifold_integrator) geodesic_move_normalized() (in module jax_rb.simulation.global_manifold_integrator) (in module jax_rb.simulation.matrix_group_integrator) geodesic_move_with_drift() (in module jax_rb.simulation.global_manifold_integrator) (in module jax_rb.simulation.matrix_group_integrator) GlobalManifold (class in jax_rb.manifolds.global_manifold) GLpLeftInvariant (class in jax_rb.manifolds.glp_left_invariant) grand() (in module jax_rb.utils.utils) Grassmann (class in jax_rb.manifolds.grassmann)

I

id_drift (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant attribute) inner() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method) inv_g_metric() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method)

ito_drift() (jax_rb.manifolds.global_manifold.GlobalManifold method) ito_move_dim_g() (in module jax_rb.simulation.matrix_group_integrator) ito_move_with_drift() (in module jax_rb.simulation.global_manifold_integrator) (in module jax_rb.simulation.matrix_group_integrator)

J

jax_rb.manifolds module jax_rb.manifolds.affine_left_invariant module jax_rb.manifolds.diag_hypersurface module jax_rb.manifolds.global_manifold module jax_rb.manifolds.glp_left_invariant module jax_rb.manifolds.grassmann module jax_rb.manifolds.matrix_left_invariant module jax_rb.manifolds.se_left_invariant module jax_rb.manifolds.sl_left_invariant module jax_rb.manifolds.so_left_invariant module

jax_rb.manifolds.spd module jax_rb.manifolds.sphere module jax_rb.manifolds.stiefel module jax_rb.simulation module jax_rb.simulation.global_manifold_integrator module jax_rb.simulation.matrix_group_integrator module jax_rb.simulation.retractive_integrator module jax_rb.simulation.simulator module jax_rb.utils module jax_rb.utils.utils module jpolar() (in module jax_rb.utils.utils)

L

laplace_beltrami() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method)

left_invariant_vector_field() (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method)

M

make_complement_basis() (in module jax_rb.utils.utils) MatrixLeftInvariant (class in jax_rb.manifolds.matrix_left_invariant) module jax_rb.manifolds jax_rb.manifolds.affine_left_invariant jax_rb.manifolds.diag_hypersurface jax_rb.manifolds.global_manifold jax_rb.manifolds.glp_left_invariant jax_rb.manifolds.grassmann jax_rb.manifolds.matrix_left_invariant jax_rb.manifolds.se_left_invariant jax_rb.manifolds.sl_left_invariant jax_rb.manifolds.so_left_invariant jax_rb.manifolds.spd jax_rb.manifolds.sphere jax_rb.manifolds.stiefel jax_rb.simulation jax_rb.simulation.global_manifold_integrator jax_rb.simulation.matrix_group_integrator jax_rb.simulation.retractive_integrator jax_rb.simulation.simulator jax_rb.utils jax_rb.utils.utils

N

name() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method)

P

PositiveDefiniteManifold (class in jax_rb.manifolds.spd) proj() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method)

pseudo_transport() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method)

R

rand_ambient() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method) rand_point() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method) rand_positive_definite() (in module jax_rb.utils.utils) rand_vec() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method) rbrownian_ito_move() (in module jax_rb.simulation.global_manifold_integrator) (in module jax_rb.simulation.matrix_group_integrator)

rbrownian_stratonovich_move() (in module jax_rb.simulation.global_manifold_integrator) (in module jax_rb.simulation.matrix_group_integrator) RealStiefelAlpha (class in jax_rb.manifolds.stiefel) retract() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method) retractive_move() (in module jax_rb.simulation.retractive_integrator) retractive_move_normalized() (in module jax_rb.simulation.retractive_integrator) run() (jax_rb.simulation.simulator.Simulator method) RunParams (class in jax_rb.simulation.simulator)

S

save_runs() (jax_rb.simulation.simulator.Simulator method) SELeftInvariant (class in jax_rb.manifolds.se_left_invariant) sigma() (jax_rb.manifolds.global_manifold.GlobalManifold method) (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method) sigma_id() (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method) simulate() (in module jax_rb.simulation.simulator) Simulator (class in jax_rb.simulation.simulator)

SLLeftInvariant (class in jax_rb.manifolds.sl_left_invariant) SOLeftInvariant (class in jax_rb.manifolds.so_left_invariant) Sphere (class in jax_rb.manifolds.sphere) stratonovich_drift() (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant method) stratonovich_move_dim_g() (in module jax_rb.simulation.matrix_group_integrator) stratonovich_move_with_drift() (in module jax_rb.simulation.global_manifold_integrator) (in module jax_rb.simulation.matrix_group_integrator)

U

unvec_skew() (in module jax_rb.utils.utils)

V

v0 (jax_rb.manifolds.matrix_left_invariant.MatrixLeftInvariant attribute)