JAX (bibliothèque Python)

JAX est une bibliothèque logicielle Python d'apprentissage automatique développée par Google et Nvidia^[2],[3]. Elle offre notamment des fonctions de dérivation automatique, de compilation à la volée et d'auto-vectorisation.

JAX

Informations

Développé par	Peter Hawkins (d), Matthew Johnson (d) et Jacob T. VanderPlas (d)
Dernière version	0.4.24 (6 février 2024)[1]
Dépôt	github.com/jax-ml/jax
Type	Bibliothèque logicielle Python (d)
Licence	Licence Apache 2.0
Site web	jax.readthedocs.io/en/latest

modifier - modifier le code - voir Wikidata (aide)