JAX

GPTKB entity

Statements (36)
Predicate Object
gptkbp:instanceOf gptkb:software
gptkbp:component gptkb:Autograd
XLA
gptkbp:developedBy gptkb:Google
gptkbp:documentation https://jax.readthedocs.io/
gptkbp:feature just-in-time compilation
random number generation
GPU support
parallelization
automatic differentiation
TPU support
custom gradients
efficient array operations
function transformations
vectorization
gptkbp:firstReleased 2018
https://www.w3.org/2000/01/rdf-schema#label JAX
gptkbp:latestReleaseVersion 0.4.25
gptkbp:license gptkb:Apache_License_2.0
gptkbp:platform gptkb:Windows
gptkb:macOS
gptkb:Linux
gptkbp:programmingLanguage gptkb:Python
gptkbp:purpose gptkb:machine_learning
numerical computing
gptkbp:relatedTo gptkb:TensorFlow
gptkb:PyTorch
gptkb:NumPy
gptkbp:repository https://github.com/google/jax
gptkbp:usedFor deep learning
scientific computing
gptkbp:bfsParent gptkb:Adam_optimizer
gptkb:Tensor
gptkb:PyTorch
gptkb:NumPy
gptkbp:bfsLayer 5