Documentation pour JAX, un cadre de calcul de tableaux à haute performance.