Back جوجل جاكس Arabic Google JAX Catalan JAX (bibliothèque Python) French JAX (ライブラリ) Japanese JAX (programmalıq támiynat) KAA JAX Korean Google JAX Russian JAX Chinese
 | This article may contain an excessive amount of intricate detail that may interest only a particular audience. (February 2025) |
| JAX | |
|---|---|
&sq=War in Ukraine&lang=en&file=File:Google_JAX_logo.svg){kind=logo} JAX logo | |
&sq=War in Ukraine&lang=en&file=File:Google_JAX_screenshot.png) | |
| Developer(s) | Google, Nvidia[1] |
| Preview release | v0.4.31
/ 30 July 2024 |
| Repository | jax on GitHub |
| Written in | Python, C++ |
| Operating system | Linux, macOS, Windows |
| Platform | Python, NumPy |
| Size | 9.0 MB |
| Type | Machine learning |
| License | Apache 2.0 |
| Website | jax |
JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning. It is developed by Google with contributions from Nvidia and other community contributors.[2][3][4]
It is described as bringing together a modified version of autograd (automatic obtaining of the gradient function through differentiation of a function) and OpenXLA's XLA (Accelerated Linear Algebra). It is designed to follow the structure and workflow of NumPy as closely as possible and works with various existing frameworks such as TensorFlow and PyTorch.[5][6] The primary features of JAX are:[7]
- Providing a unified NumPy-like interface to computations that run on CPU, GPU, or TPU, in local or distributed settings.
- Built-in Just-In-Time (JIT) compilation via Open XLA, an open-source machine learning compiler ecosystem.
- Efficient evaluation of gradients via its automatic differentiation transformations.
- Automatically vectorized to efficiently map them over arrays representing batches of inputs.
- ^ "jax/AUTHORS at main · jax-ml/jax". GitHub. Retrieved December 21, 2024.
- ^ Bradbury, James; Frostig, Roy; Hawkins, Peter; Johnson, Matthew James; Leary, Chris; MacLaurin, Dougal; Necula, George; Paszke, Adam; Vanderplas, Jake; Wanderman-Milne, Skye; Zhang, Qiao (2022-06-18), "JAX: Autograd and XLA", Astrophysics Source Code Library, Google, Bibcode:2021ascl.soft11002B, archived from the original on 2022-06-18, retrieved 2022-06-18
- ^ Frostig, Roy; Johnson, Matthew James; Leary, Chris (2018-02-02). "Compiling machine learning programs via high-level tracing" (PDF). MLsys: 1–3. Archived (PDF) from the original on 2022-06-21.
{{cite journal}}: CS1 maint: date and year (link) - ^ "Using JAX to accelerate our research". www.deepmind.com. Archived from the original on 2022-06-18. Retrieved 2022-06-18.
- ^ Lynley, Matthew. "Google is quietly replacing the backbone of its AI product strategy after its last big push for dominance got overshadowed by Meta". Business Insider. Archived from the original on 2022-06-21. Retrieved 2022-06-21.
- ^ "Why is Google's JAX so popular?". Analytics India Magazine. 2022-04-25. Archived from the original on 2022-06-18. Retrieved 2022-06-18.
- ^ "Quickstart — JAX documentation".
© MMXXIII Rich X Search. We shall prevail. All rights reserved. Rich X Search