The journal
Open the journal →The conversation starts in the journal — be the first to post.
About JAX
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.
Everything about JAX →- Website
- https://jax.readthedocs.io/en/latest/