Index

A

approx_nearest() (jax_par_trans.manifolds.flag.Flag method) (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.stiefel.Stiefel method)

asym() (in module jax_par_trans.expv.utils) (in module par_trans.utils.utils)

C

christoffel_gamma() (jax_par_trans.manifolds.flag.Flag method) (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.so_alpha.SOAlpha method) (par_trans.manifolds.stiefel.Stiefel method)

christoffel_gamma_lie() (par_trans.manifolds.so_alpha.SOAlpha method) cz() (in module jax_par_trans.expv.utils) (in module par_trans.utils.utils)

D

dexp() (jax_par_trans.manifolds.flag.Flag method) (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.so_alpha.SOAlpha method) (par_trans.manifolds.stiefel.Stiefel method)

dot() (jax_par_trans.expv.expv.LinearOperator method)

E

exp() (jax_par_trans.manifolds.flag.Flag method) (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.so_alpha.SOAlpha method) (par_trans.manifolds.stiefel.Stiefel method)

expm_multiply() (in module par_trans.utils.expm_multiply_np) expv() (jax_par_trans.expv.expv.LinearOperator method)

F

Flag (class in jax_par_trans.manifolds.flag) (class in par_trans.manifolds.flag)

FlagCanonicalParallelOperator (class in jax_par_trans.manifolds.flag)

G

GLpBeta (class in par_trans.manifolds.glp_beta)

grand() (in module jax_par_trans.expv.utils)

H

hcat() (in module jax_par_trans.expv.utils) (in module par_trans.utils.utils)

I

inner() (jax_par_trans.manifolds.flag.Flag method) (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.so_alpha.SOAlpha method) (par_trans.manifolds.stiefel.Stiefel method)

J

jax_par_trans.expv module jax_par_trans.expv.expv module jax_par_trans.expv.utils module

jax_par_trans.manifolds module jax_par_trans.manifolds.flag module jax_par_trans.manifolds.stiefel module

L

lie() (in module jax_par_trans.expv.utils) (in module par_trans.utils.utils)

LinearOperator (class in jax_par_trans.expv.expv)

M

make_ar() (jax_par_trans.manifolds.flag.Flag method) (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.stiefel.Stiefel method) module jax_par_trans.expv jax_par_trans.expv.expv jax_par_trans.expv.utils jax_par_trans.manifolds jax_par_trans.manifolds.flag jax_par_trans.manifolds.stiefel par_trans.manifolds par_trans.manifolds.flag par_trans.manifolds.glp_beta par_trans.manifolds.so_alpha par_trans.manifolds.stiefel par_trans.utils par_trans.utils.expm_multiply_np par_trans.utils.utils

N

name() (jax_par_trans.manifolds.flag.Flag method) (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.so_alpha.SOAlpha method) (par_trans.manifolds.stiefel.Stiefel method)

O

one_norm_est() (jax_par_trans.expv.expv.LinearOperator method)

P

par_bal() (in module par_trans.manifolds.stiefel) par_trans.manifolds module par_trans.manifolds.flag module par_trans.manifolds.glp_beta module par_trans.manifolds.so_alpha module par_trans.manifolds.stiefel module par_trans.utils module par_trans.utils.expm_multiply_np module

par_trans.utils.utils module parallel() (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.so_alpha.SOAlpha method) (par_trans.manifolds.stiefel.Stiefel method) parallel_canonical() (jax_par_trans.manifolds.flag.Flag method) (par_trans.manifolds.flag.Flag method) parallel_canonical_expm_multiply() (par_trans.manifolds.flag.Flag method) parallel_expm_multiply() (par_trans.manifolds.stiefel.Stiefel method) proj() (jax_par_trans.manifolds.flag.Flag method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.so_alpha.SOAlpha method) proj_m() (jax_par_trans.manifolds.flag.Flag method) (par_trans.manifolds.flag.Flag method)

R

rand_ambient() (jax_par_trans.manifolds.flag.Flag method) (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.so_alpha.SOAlpha method) (par_trans.manifolds.stiefel.Stiefel method) rand_point() (jax_par_trans.manifolds.flag.Flag method) (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.so_alpha.SOAlpha method) (par_trans.manifolds.stiefel.Stiefel method)

rand_vec() (jax_par_trans.manifolds.flag.Flag method) (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.so_alpha.SOAlpha method) (par_trans.manifolds.stiefel.Stiefel method) retract() (jax_par_trans.manifolds.flag.Flag method) (jax_par_trans.manifolds.stiefel.Stiefel method) (par_trans.manifolds.flag.Flag method) (par_trans.manifolds.glp_beta.GLpBeta method) (par_trans.manifolds.so_alpha.SOAlpha method) (par_trans.manifolds.stiefel.Stiefel method)

S

set_params() (jax_par_trans.expv.expv.LinearOperator method) (jax_par_trans.manifolds.flag.FlagCanonicalParallelOperator method) (jax_par_trans.manifolds.stiefel.StiefelParallelOperator method) SOAlpha (class in par_trans.manifolds.so_alpha) solve_w() (in module par_trans.manifolds.flag) (in module par_trans.manifolds.stiefel)

Stiefel (class in jax_par_trans.manifolds.stiefel) (class in par_trans.manifolds.stiefel) StiefelParallelOperator (class in jax_par_trans.manifolds.stiefel) sym() (in module jax_par_trans.expv.utils) (in module par_trans.utils.utils) symf() (jax_par_trans.manifolds.flag.Flag method) (par_trans.manifolds.flag.Flag method)

V

vcat() (in module jax_par_trans.expv.utils) (in module par_trans.utils.utils)