https://github.com/deskiziarecords/tungsten-alpha

Production JAX framework eliminating dynamic shape recompilation. 7-layer Adelic Riemannian stack: NRRB (threshold bridges), CSM (HLO hardening), ATC (adaptive JIT), ARSM (pure state), ASTC (telemetry), AFRH (FIM/nat-grad), AFRC (cluster transport). 100% JIT-stable, scales 10k+ Orbax→Triton ready. Tungsten Alpha OS

custom-jax-primitives fisher-information-matrix hlo-fusion jax jit-compilation natural-gradient orbax riemannian-optimization sensor-swarms tpu triton-inference xla

Last synced: 23 Apr 2026