Python
Comment transformer du code NumPy familier en code GPU compilable
JAX combine la syntaxe familière de NumPy avec la différentiation automatique et la compilation XLA, transformant Python en code accéléré par GPU. Voici comment cela fonctionne.