II.
Library overview
Reference · livelibrary:jax
JAX overview
NumPy-compatible array library with grad/jit/vmap/pmap; backbone of Flax and many high-performance ML research codebases.
Attributes
displayName
JAX
belongsToLanguageId
description
NumPy-compatible array library with grad/jit/vmap/pmap; backbone of
Flax and many high-performance ML research codebases.
Outgoing edges
belongs_to_language1
- language:python·LanguagePython
library_used_by1
- skill-area:machine-learning·SkillAreaMachine Learning
Incoming edges
alternative_to1
- library:pytorch·LibraryPyTorch