JAX: como acelerar computação numérica e modelos de IA em Python
JAX combina uma API parecida com NumPy e transformações para compilação, diferenciação automática, vetorização e paralelismo. Veja como começar e quando ele faz sentido para computação científica e IA.