JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning. It is developed by Google with contributions from Nvidia and other community contributors.

From Wikipedia (CC BY-SA 4.0).

On IJR

No public IJR answers mention JAX yet. As people ask about it and share their answers, they'll appear here.

Facts from Wikidata (CC0).

Not who you're looking for? Others named JAX: American drag queen · commune in Haute-Loire, France · Mortal Kombat character · American singer and songwriter

Overview

From Wikipedia (CC BY-SA 4.0).