Building with AI

JAX

Reference entryDeep learning frameworks

Reference entryJAX

JAX is a Python numerical-computing library that adds automatic differentiation, compilation, vectorization, and accelerator support to NumPy-style programs.

More in Building with AI Browse all terms

Where it sits