Type to search…

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.

Introducció

JAX

You're reading a preview.

Sign in with Google to read the full page. Follow the learning path — each page opens once you've read the ones it builds on. Students and teachers read their course pages without limit.

Sign in