Back to jax

Hero

docs/hero.html

0.3.25215 B
Original Source

High performance array computing

JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning.