File size: 1,825 Bytes
79eb85c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | # Copyright 2024-2025 AI Whisperers (https://github.com/Ai-Whisperers)
#
# Licensed under the PolyForm Noncommercial License 1.0.0
# See LICENSE file in the repository root for full license text.
#
# For commercial licensing inquiries: support@aiwhisperers.com
"""Geometry module for hyperbolic and p-adic operations.
This module provides geoopt-backed implementations of hyperbolic geometry
operations, providing numerical stability and optimized performance.
Key components:
- PoincareBall: Manifold operations via geoopt
- Stable exp/log maps with automatic edge-case handling
- RiemannianAdam optimizer support
"""
from .poincare import (ManifoldParameter, ManifoldTensor, PoincareModule,
RiemannianAdam, RiemannianSGD,
create_manifold_parameter, create_manifold_tensor,
exp_map_zero, get_manifold, get_riemannian_optimizer,
lambda_x, log_map_zero, mobius_add, parallel_transport,
poincare_distance, poincare_distance_matrix,
project_to_poincare)
from .holographic_poincare import (
BoundaryPoint,
HolographicLoss,
HolographicPoincareManifold,
HolographicProjection,
)
__all__ = [
# Core Poincare operations
"get_manifold",
"poincare_distance",
"poincare_distance_matrix",
"project_to_poincare",
"exp_map_zero",
"log_map_zero",
"mobius_add",
"lambda_x",
"parallel_transport",
"PoincareModule",
"create_manifold_parameter",
"create_manifold_tensor",
"get_riemannian_optimizer",
"ManifoldParameter",
"ManifoldTensor",
"RiemannianAdam",
"RiemannianSGD",
# Holographic extensions
"HolographicPoincareManifold",
"HolographicProjection",
"HolographicLoss",
"BoundaryPoint",
]
|