SOURCE-LINKED INTELLIGENCE

hyperbolix: Hyperbolic Deep Learning in JAX

arXiv · AI, language, vision and robotics · article · Sep 23, 2026 · UTC

We present hyperbolix, an open-source library for hyperbolic deep learning in JAX, built on Flax NNX. To our knowledge, it is the first comprehensive, general-purpose hyperbolic deep learning library in JAX. It includes six manifolds with a common interface: Euclidean space, the Poincaré ball, the hyperboloid, the $κ$-stereographic model, mixed-curvature product spaces, and the proper velocity space. We implement layer families that cover linear layers, convolutions, attention, normalization, positional encoding, regression, and vector quantization. These building blocks span methods ranging f

Read original source ↗ Open in workspace

recordType

paper

region

Global