JAX
Eine Python-Bibliothek für Hochleistungs-Maschinelles Lernen und numerische Berechnungen, die von Google entwickelt wurde. Sie kombiniert eine NumPy-ähnliche API mit Autograd (automatischer Differenzierung) und XLA (Accelerated Linear Algebra) für die Kompilierung und Ausführung auf CPUs, GPUs und TPUs.