Skalierbare und differenzierbare föderierte Primitive in JAX: FAX, eine leistungsfähige Bibliothek für verteilte und föderierte Berechnungen

FAX ist eine JAX-basierte Bibliothek, die große verteilte und föderierte Berechnungen in Rechenzentren und geräteübergreifenden Anwendungen unterstützt. FAX integriert Bausteine für föderierte Berechnungen als Primitive in JAX, was drei Hauptvorteile bietet: Übersetzung in XLA HLO, Implementierung der Föderierten Automatischen Differenzierung und Interpretation in bestehende Produktions-Föderations-Computesysteme.