Statements (24)
| Predicate | Object |
|---|---|
| gptkbp:instanceOf |
gptkb:model
gptkb:software |
| gptkbp:category |
deep learning
scientific computing |
| gptkbp:developedBy |
gptkb:Google
|
| gptkbp:documentation |
https://jax.readthedocs.io/
|
| gptkbp:feature |
just-in-time compilation
function transformations vectorization |
| gptkbp:firstReleased |
2018
|
| gptkbp:latestReleaseVersion |
2024
|
| gptkbp:license |
gptkb:Apache_License_2.0
|
| gptkbp:programmingLanguage |
gptkb:Python
|
| gptkbp:relatedTo |
gptkb:TensorFlow
gptkb:PyTorch |
| gptkbp:repository |
https://github.com/google/jax
|
| gptkbp:supports |
GPU acceleration
automatic differentiation TPU acceleration |
| gptkbp:uses |
gptkb:NumPy_API
XLA (Accelerated Linear Algebra) |
| gptkbp:bfsParent |
gptkb:Google_Brain_(former)
|
| gptkbp:bfsLayer |
7
|
| https://www.w3.org/2000/01/rdf-schema#label |
JAX: Autograd and XLA
|