#
orbax
Here are 2 public repositories matching this topic...
Tungsten Alpha: A JAX-native, 7-layer Adelic-Riemannian Operating System. Implements resurgent gradient flow through discrete logic and Levi-Civita parallel transport for decentralized XLA clusters. Hardened for zero-jitter, systolic array execution.
high-performance-computing artificial-general-intelligence tpu jax riemannian-geometry riemannian-optimization fisher-information-matrix xla information-geometry jit-compilation natural-gradient parallel-transport machine-learning-infrastructure triton-inference orbax custom-jax-primitives hlo-fusion sensor-swarms manifold-learning-decentralized-ai
-
Updated
Mar 16, 2026 - Python
Add this topic to your repo
To associate your repository with the orbax topic, visit your repo's landing page and select "manage topics."