#
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
Improve this page
Add a description, image, and links to the orbax topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the orbax topic, visit your repo's landing page and select "manage topics."