Explore relationships

JAX (software)

JAX is a Pythonlibrary 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.

Use + to expand a branch. Click a topic name to open its summary.