Dokumentation für JAX, ein leistungsstarkes Array-Computing-Framework.