JAX is a Python-based toolset (Autograd and XLA) for high performance machine learning applications. Features familiar API, transformations and multiple backend support.
To prevent this message from showing again, please enable pop-up blockers for support.broadcom.com or click Continue to proceed.