Properties (48)
| Predicate | Object |
|---|---|
| gptkbp:instanceOf |
software library
|
| gptkbp:developedBy |
gptkb:Google
|
| gptkbp:hasAmenities |
official website
|
| gptkbp:hasFeature |
just-in-time compilation
parallelization vectorization jax.lax jax.numpy jax.random jax.scipy jax.custom_jvp jax.custom_vjp jax.device_get jax.device_put jax.grad jax.jaxpr jax.jit jax.make_jaxpr jax.pure jax.vmap NumPy compatibility jax.custom_transforms custom_gradients jax.experimental jax.pmap |
| gptkbp:hasOccupation |
open-source contributors
|
| gptkbp:hasRelatedPatent |
image processing
deep learning natural language processing neural networks scientific computing statistical modeling reinforcement learning optimization problems probabilistic programming |
| https://www.w3.org/2000/01/rdf-schema#label |
JAX
|
| gptkbp:isCompatibleWith |
gptkb:PyTorch
TensorFlow NumPy |
| gptkbp:isPartOf |
TensorFlow ecosystem
|
| gptkbp:isUsedBy |
data scientists
researchers |
| gptkbp:language |
Python
|
| gptkbp:provides |
automatic differentiation
|
| gptkbp:releaseDate |
2018
|
| gptkbp:supports |
GPU_acceleration
TPU_acceleration |
| gptkbp:usedFor |
machine learning
|