PyPI page
Home page
Author:
JAX team
License:
Apache-2.0
Summary:
Differentiate, compile, and transform Numpy code.
Latest version:
0.4.30
Required dependencies:
importlib-metadata
|
jaxlib
|
ml-dtypes
|
numpy
|
opt-einsum
|
scipy
Optional dependencies:
jax-cuda12-plugin
|
jaxlib
|
libtpu-nightly
|
requests
Downloads last day:
128,975
Downloads last week:
754,756
Downloads last month:
3,480,791