Documentación sobre JAX, un marco informático de alto rendimiento para matrices.