PyPI page
Home page
Author:
JAX team
License:
Apache-2.0
Summary:
Differentiate, compile, and transform Numpy code.
Latest version:
0.10.2
Required dependencies:
frxlib
|
ml-dtypes
|
numpy
|
opt-einsum
|
scipy
|
zk-dtypes
Optional dependencies:
frx-cuda12-plugin
|
frx-cuda13-plugin
|
frx-rocm7-plugin
|
frxlib
|
kubernetes
|
libtpu
|
requests
|
xprof
Downloads last day:
1
Downloads last week:
19
Downloads last month:
60
7-day smoothing