#
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-computingartificial-general-intelligencetpujaxriemannian-geometryriemannian-optimizationfisher-information-matrixxlainformation-geometryjit-compilationnatural-gradientparallel-transportmachine-learning-infrastructuretriton-inferenceorbaxcustom-jax-primitiveshlo-fusionsensor-swarmsmanifold-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."