JAX (bibliothèque Python)
JAX
| Développé par | Peter Hawkins (d), Matthew Johnson (d) et Jacob T. VanderPlas (d) |
|---|---|
| Dernière version | 0.4.24 ()[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 |
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
- ↑ « https://github.com/google/jax/releases/tag/jax-v0.4.24 »
- ↑ (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])
- ↑ (en) Grigory Sapunov, Deep Learning with JAX, Simon & Schuster, (ISBN 978-1-63343-888-0, lire en ligne)
Liens externes
- Portail de la programmation informatique
- Portail de Google
- Portail de l’intelligence artificielle