Share to: share facebook share twitter share wa share telegram print page

JAX (bibliothèque Python)

JAX
Description de l'image Google JAX logo.svg.

Informations
Développé par Peter Hawkins (d), Matthew Johnson (d) et Jacob T. VanderPlas (d)Voir et modifier les données sur Wikidata
Dernière version 0.4.24 ()[1]Voir et modifier les données sur Wikidata
Dépôt github.com/jax-ml/jaxVoir et modifier les données sur Wikidata
Type Bibliothèque logicielle Python (d)Voir et modifier les données sur Wikidata
Licence Licence Apache 2.0Voir et modifier les données sur Wikidata
Site web jax.readthedocs.io/en/latestVoir et modifier les données sur Wikidata

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.

Notes et références

  1. « https://github.com/google/jax/releases/tag/jax-v0.4.24 »
  2. (en) Roy Frostig, Matthew James Johnson et Chris Leary, « Compiling machine learning programs via high-level tracing », Systems for Machine Learning, vol. 4, no 9,‎ (lire en ligne [PDF])
  3. (en) Grigory Sapunov, Deep Learning with JAX, Simon & Schuster, (ISBN 978-1-63343-888-0, lire en ligne)

Liens externes

Kembali kehalaman sebelumnya