PyPI page
Home page
Author:
JAX team
License:
Apache-2.0
Summary:
Differentiate, compile, and transform Numpy code.
Latest version:
0.9.0.1
Required dependencies:
jaxlib
|
ml_dtypes
|
numpy
|
opt_einsum
|
scipy
Optional dependencies:
jax-cuda12-plugin
|
jax-cuda13-plugin
|
jax-rocm7-plugin
|
jaxlib
|
kubernetes
|
libtpu
|
requests
|
xprof
Downloads last day:
357,477
Downloads last week:
4,530,029
Downloads last month:
17,482,072