Tree-Preconditioned Differentiable Optimization and Axioms as Layers. - Yuexin Liao (CoRR, 2026)