Atlas Graph Explorer
Wiki
Graph
Edges
Home
Library
library:jax
JAX
library:jax
Library
domain/libraries/libraries-python.yaml
·
Open in Graph →
overview
json
graph
Attributes
displayName
JAX
belongsToLanguageId
language:python
description
NumPy-compatible array library with grad/jit/vmap/pmap; backbone of Flax and many high-performance ML research codebases.
Outgoing edges
(1)
belongs_to_language
1
language:python
Incoming edges
(0)
None.