openteamsinc
pypi/jax
Risk Profile
Package
Source
Risk Profile
Maturity:
Mature
Health:
High Risk
published package has a different name than specified in the source code
Legal:
Caution Needed
License has additional text that may require review before use
License includes a patent grant clause
Security:
Healthy
Package
pypi
jax
Version
0.11.0
Last Release Date
6 days ago
License
Apache
Dependencies
jaxlib
<=0.11.0, >=0.11.0
ml_dtypes
>=0.5.0
numpy
>=2.1
opt_einsum
scipy
>=1.15
minimum-jaxlib:
jaxlib
==0.11.0
ci:
jaxlib
==0.10.2
tpu:
jaxlib
<=0.11.0, >=0.11.0
libtpu
==0.0.44.*
requests
cuda:
jaxlib
<=0.11.0, >=0.11.0
jax-cuda12-plugin
[with-cuda]
<=0.11.0, >=0.11.0
cuda12:
jaxlib
<=0.11.0, >=0.11.0
jax-cuda12-plugin
[with-cuda]
<=0.11.0, >=0.11.0
cuda13:
jaxlib
<=0.11.0, >=0.11.0
jax-cuda13-plugin
[with-cuda]
<=0.11.0, >=0.11.0
cuda12-local:
jaxlib
<=0.11.0, >=0.11.0
jax-cuda12-plugin
<=0.11.0, >=0.11.0
cuda13-local:
jaxlib
<=0.11.0, >=0.11.0
jax-cuda13-plugin
<=0.11.0, >=0.11.0
rocm7-local:
jaxlib
<=0.11.0, >=0.11.0
jax-rocm7-plugin
==0.11.0.*
oneapi:
jaxlib
<=0.11.0, >=0.11.0
jax-oneapi-plugin
[with-oneapi]
<=0.11.0, >=0.11.0
k8s:
kubernetes
xprof:
xprof
Source
Location
jax-ml/jax
Last Source Update
about 23 hours ago
Licenses
Apache License 2.0
Mozilla Public License 2.0
(text added)
BSD 3-Clause "New" or "Revised" License
BSD 3-Clause "New" or "Revised" License
Distribution Destinations
pypi/jax-ffi-example