Index

A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | R | S | T | W

A

AdaptiveArmijo (class in geojax.optimization)
AdaptiveRegularizationCubics (class in geojax.optimization)
adjoint_jacobian() (geojax.optimization.LeastSquares method)

agglomerative_clustering() (in module geojax.learning)
align() (geojax.geometry.KendallShape method)
AlternatingGradient (class in geojax.optimization)
as_manifold_data() (in module geojax.learning)

B

BacktrackingArmijo (class in geojax.optimization)
BarycentricCodingResult (class in geojax.learning)
BarzilaiBorwein (class in geojax.optimization)
batch_cost_and_grad() (geojax.optimization.FiniteSum method)

belongs() (geojax.geometry.base.ManifoldProtocol method)
- (geojax.geometry.Sphere method)
biswas_ghosh_two_sample_test() (in module geojax.learning)
bootstrap_frechet_mean() (in module geojax.learning)
BootstrapResult (class in geojax.learning)

C

check_manifold_data() (in module geojax.learning)
chordal_dist() (geojax.geometry.GrassmannProjection method)
- (geojax.geometry.SphereExtrinsic method)
classical_mds() (in module geojax.learning)
ClusteringResult (class in geojax.learning)
competitive_quantization() (in module geojax.learning)
ConjugateGradient (class in geojax.optimization)

ConstantSchedule (class in geojax.optimization)
ConstantStep (class in geojax.optimization)
converged (geojax.geometry.StiefelLogInfo attribute)
CoresetResult (class in geojax.learning)
CorrelationAffineQuotient (class in geojax.geometry)
CorrelationECM (class in geojax.geometry)
CorrelationLEC (class in geojax.geometry)
CosineDecay (class in geojax.optimization)

D

DataValidationReport (class in geojax.learning)
DictionaryLearningResult (class in geojax.learning)
dim (geojax.geometry.Sphere property)
dist() (geojax.geometry.base.GeometryProtocol method)

dist_batch() (geojax.geometry.base.GeometryMixin method)
dist_is_exact (geojax.geometry.Product property)

E

egrad2rgrad() (geojax.geometry.Stiefel method)
egrad_to_rgrad() (geojax.geometry.base.ManifoldProtocol method)
- (geojax.geometry.Sphere method)
- (geojax.geometry.Stiefel method)
ehess_to_rhess() (geojax.geometry.base.GeometryMixin method)
ehess_vec() (geojax.optimization.Minimize method)
Elliptope (class in geojax.geometry)
embed() (geojax.geometry.GrassmannProjection method)
- (geojax.geometry.SphereExtrinsic method)
- (geojax.learning.EquivariantEmbeddingProtocol method)
embed_tangent() (geojax.geometry.GrassmannProjection method)
embedding_dist() (geojax.geometry.SphereExtrinsic method)
EmbeddingResult (class in geojax.learning)
empirical_wasserstein_distance() (in module geojax.learning)

EnclosingBallResult (class in geojax.learning)
energy_two_sample_test() (in module geojax.learning)
EquivariantEmbeddingProtocol (class in geojax.learning)
Euclidean (class in geojax.geometry)
ExactGeometryMixin (class in geojax.geometry.base)
exp() (geojax.geometry.base.GeometryProtocol method)
exp_batch() (geojax.geometry.base.GeometryMixin method)
exp_is_exact (geojax.geometry.Product property)
extrinsic_mean() (geojax.geometry.GrassmannProjection method)
- (geojax.geometry.SphereExtrinsic method)

F

FiniteSum (class in geojax.optimization)
FixedRank (class in geojax.geometry)
frechet_anova() (in module geojax.learning)
frechet_mean() (in module geojax.learning)
frechet_median() (in module geojax.learning)
FrechetMeanResult (class in geojax.learning)

FrechetMedianResult (class in geojax.learning)
from_embedding() (geojax.geometry.GrassmannProjection method)
- (geojax.geometry.SphereExtrinsic method)
from_frame() (geojax.geometry.GrassmannProjection method)
from_poincare() (geojax.geometry.Hyperboloid method)
from_projector_tangent() (geojax.geometry.GrassmannProjection method)

G

GaussNewton (class in geojax.optimization)
GeneralizedGrassmann (class in geojax.geometry)
GeneralizedStiefel (class in geojax.geometry)
geodesic_barycentric_coding() (in module geojax.learning)
geodesic_dist() (geojax.geometry.GrassmannProjection method)
geodesic_flow() (geojax.geometry.Sphere method)
geodesic_interpolation() (in module geojax.learning)
geodesic_m_estimator() (in module geojax.learning)
geodesic_regression() (in module geojax.learning)

geodesic_spatial_depth() (in module geojax.learning)
GeodesicRegressionModel (class in geojax.learning)
GeometryMixin (class in geojax.geometry.base)
GeometryProtocol (class in geojax.geometry.base)
Grassmann (class in geojax.geometry)
GrassmannProjection (class in geojax.geometry)
group_exp() (geojax.geometry.SpecialEuclidean method)
- (geojax.geometry.SpecialOrthogonal method)
group_log() (geojax.geometry.SpecialEuclidean method)
- (geojax.geometry.SpecialOrthogonal method)

H

hessian_conversion_is_exact (geojax.geometry.Product property)
hessian_operator() (geojax.optimization.Minimize method)
HierarchicalClusteringResult (class in geojax.learning)

horizontal_lift() (geojax.geometry.CorrelationAffineQuotient method)
Hyperboloid (class in geojax.geometry)
HypothesisTestResult (class in geojax.learning)

I

InfoEntry (class in geojax.optimization)
inner() (geojax.geometry.base.ManifoldProtocol method)
- (geojax.geometry.Sphere method)
invretr() (geojax.geometry.base.GeometryMixin method)

is_tangent() (geojax.geometry.base.ManifoldProtocol method)
- (geojax.geometry.Sphere method)
isomap() (in module geojax.learning)
iterations (geojax.geometry.StiefelLogInfo attribute)

J

jacobian_vec() (geojax.optimization.LeastSquares method)

K

k (geojax.geometry.Stiefel property)
- (geojax.geometry.StiefelEuclidean property)
KendallShape (class in geojax.geometry)
kernel_mmd_two_sample_test() (in module geojax.learning)
kernel_pca() (in module geojax.learning)
kernel_regression() (in module geojax.learning)

KernelCVResult (class in geojax.learning)
KernelRegressionModel (class in geojax.learning)
kmeans() (in module geojax.learning)
kmedoids() (in module geojax.learning)
KNearestNeighborsModel (class in geojax.learning)
knn_classifier() (in module geojax.learning)

L

label_propagation() (in module geojax.learning)
LBFGS (class in geojax.optimization)
LearningCapabilityError
LeastSquares (class in geojax.optimization)
LevenbergMarquardt (class in geojax.optimization)
lightweight_coreset() (in module geojax.learning)
lincomb() (geojax.geometry.base.GeometryMixin method)
LineSearchProtocol (class in geojax.optimization)
LineSearchResult (class in geojax.optimization)
LineSearchState (class in geojax.optimization)
LineSearchStats (class in geojax.optimization)
local_polynomial_regression() (in module geojax.learning)

LocalPolynomialRegressionModel (class in geojax.learning)
log() (geojax.geometry.base.GeometryProtocol method)
log_batch() (geojax.geometry.base.GeometryMixin method)
log_is_exact (geojax.geometry.Product property)
log_with_info() (geojax.geometry.GeneralizedStiefel method)
- (geojax.geometry.Stiefel method)
- (geojax.geometry.StiefelEuclidean method)

M

manifold_dictionary_learning() (in module geojax.learning)
manifold_regularized_regression() (in module geojax.learning)
ManifoldData (class in geojax.learning)
ManifoldDataAdapterProtocol (class in geojax.learning)
ManifoldProtocol (class in geojax.geometry.base)
mean_shift() (in module geojax.learning)

metric_distance_ranks() (in module geojax.learning)
MetricLearningModel (class in geojax.learning)
MetricRanksResult (class in geojax.learning)
minibatch_frechet_mean() (in module geojax.learning)
minibatch_kmeans() (in module geojax.learning)
Minimize (class in geojax.optimization)
minimum_enclosing_ball() (in module geojax.learning)

N

n (geojax.geometry.Stiefel property)
- (geojax.geometry.StiefelEuclidean property)
nearest_centroid_classifier() (in module geojax.learning)
nearest_neighbors() (in module geojax.learning)
NearestCentroidModel (class in geojax.learning)
NeighborsResult (class in geojax.learning)
NelderMead (class in geojax.optimization)

NewtonCG (class in geojax.optimization)
norm() (geojax.geometry.base.ManifoldProtocol method)
- (geojax.geometry.Sphere method)
normal_operator() (geojax.optimization.LeastSquares method)
normalize() (geojax.geometry.GrassmannProjection method)

O

Oblique (class in geojax.geometry)
operation_kind() (geojax.geometry.base.GeometryMixin method)

optimal_transport_map() (geojax.geometry.SPDBuresWasserstein method)

P

pair_mean() (geojax.geometry.base.GeometryMixin method)
paired_frechet_test() (in module geojax.learning)
pairwise_distances() (in module geojax.learning)
ParticleSwarm (class in geojax.optimization)
phate() (in module geojax.learning)
PoincareBall (class in geojax.geometry)

PolynomialDecay (class in geojax.optimization)
principal_geodesic_analysis() (in module geojax.learning)
ProbabilitySimplex (class in geojax.geometry)
Product (class in geojax.geometry)
proj() (geojax.geometry.GrassmannProjection method)
project() (geojax.geometry.base.ManifoldProtocol method)
project_embedding() (geojax.geometry.GrassmannProjection method)
- (geojax.geometry.SphereExtrinsic method)
projection() (geojax.geometry.GrassmannProjection method)
projection_dist() (geojax.geometry.GrassmannProjection method)
projector() (geojax.geometry.Grassmann method)

R

random_point() (geojax.geometry.base.ManifoldProtocol method)
- (geojax.geometry.Sphere method)
random_tangent() (geojax.geometry.base.ManifoldProtocol method)
- (geojax.geometry.Sphere method)
RankKPSD (class in geojax.geometry)
RankKPSDBuresWasserstein (class in geojax.geometry)
register_manifold_data_adapter() (in module geojax.learning)
residual_norm (geojax.geometry.StiefelLogInfo attribute)
residual_norm() (geojax.optimization.LeastSquares method)
residual_value() (geojax.optimization.LeastSquares method)
retr() (geojax.geometry.base.GeometryMixin method)

RetractionGeometryMixin (class in geojax.geometry.base)
rhess_vec() (geojax.optimization.Minimize method)
riemannian_gradient_jvp_is_exact (geojax.geometry.Product property)
riemannian_metric_learning() (in module geojax.learning)
RobustLocationResult (class in geojax.learning)

S

sammon_mapping() (in module geojax.learning)
sample_batch() (geojax.optimization.FiniteSum method)
select_kernel_bandwidth() (in module geojax.learning)
SemiSupervisedResult (class in geojax.learning)
shape (geojax.geometry.Sphere property)
sinkhorn_divergence() (in module geojax.learning)
solve() (geojax.optimization.ConjugateGradient method)
- (geojax.optimization.Minimize method)
- (geojax.optimization.SteepestDescent method)
SPDAffineInvariant (class in geojax.geometry)
SPDBuresWasserstein (class in geojax.geometry)
SPDLogEuclidean (class in geojax.geometry)
SpecialEuclidean (class in geojax.geometry)
SpecialOrthogonal (class in geojax.geometry)
Spectrahedron (class in geojax.geometry)
spectral_clustering() (in module geojax.learning)
Sphere (class in geojax.geometry)
SphereExtrinsic (class in geojax.geometry)
split_key() (geojax.optimization.Minimize method)
squared_chordal_dist() (geojax.geometry.GrassmannProjection method)
- (geojax.geometry.SphereExtrinsic method)
squared_dist() (geojax.geometry.base.GeometryMixin method)

SteepestDescent (class in geojax.optimization)
step_norm (geojax.geometry.StiefelLogInfo attribute)
StepScheduleProtocol (class in geojax.optimization)
Stiefel (class in geojax.geometry)
StiefelEuclidean (class in geojax.geometry)
StiefelLogInfo (class in geojax.geometry)
StochasticGradient (class in geojax.optimization)
streaming_frechet_mean() (in module geojax.learning)
StrongWolfe (class in geojax.optimization)
sylvester() (geojax.geometry.SPDBuresWasserstein method)

T

tangent_project() (geojax.geometry.base.ManifoldProtocol method)
tangent_space_discriminant_analysis() (in module geojax.learning)
tangent_space_logistic_regression() (in module geojax.learning)
tangent_space_map() (in module geojax.learning)
TangentFeatureMap (class in geojax.learning)
TangentSpaceClassifierModel (class in geojax.learning)
to_embedding() (geojax.geometry.SphereExtrinsic method)
to_frame() (geojax.geometry.GrassmannProjection method)
to_poincare() (geojax.geometry.Hyperboloid method)
to_tangent() (geojax.geometry.GrassmannProjection method)
Torus (class in geojax.geometry)
transp() (geojax.geometry.base.RetractionGeometryMixin method)

transport() (geojax.geometry.base.ManifoldProtocol method)
transport_is_isometric (geojax.geometry.Product property)
transport_is_parallel (geojax.geometry.Product property)
TransportResult (class in geojax.learning)
trimmed_frechet_mean() (in module geojax.learning)
TrustRegions (class in geojax.optimization)
tsne() (in module geojax.learning)

W

wasserstein_two_sample_test() (in module geojax.learning)

wrap_angles() (in module geojax.geometry.torus)