dev-python/jax
Differentiate, compile, and transform Numpy code.
-
jax-0.9.0~amd64 ~x86ci cuda cuda12 cuda12-local cuda13 cuda13-local k8s minimum-jaxlib rocm tpu xprof python_targets_python3_11 python_targets_python3_12 python_targets_python3_13 python_targets_python3_14
View
Download
Browse License: Apache-2.0 Overlay: pypi -
jax-0.8.3~amd64 ~x86ci cuda cuda12 cuda12-local cuda13 cuda13-local k8s minimum-jaxlib rocm tpu xprof python_targets_python3_11 python_targets_python3_12 python_targets_python3_13 python_targets_python3_14
View
Download
Browse License: Apache-2.0 Overlay: pypi -
jax-0.4.25~amd64 ~arm64 ~x86python_targets_python3_12 python_targets_python3_13 python_targets_python3_14
View
Download
Browse License: Overlay: blshkv-overlay
USE Flags
ci
* This flag is undocumented *
cuda
Global: Build CUDA binaries.
cuda12
* This flag is undocumented *
cuda12-local
* This flag is undocumented *
cuda13
* This flag is undocumented *
cuda13-local
* This flag is undocumented *
k8s
* This flag is undocumented *
minimum-jaxlib
* This flag is undocumented *
rocm
* This flag is undocumented *
tpu
* This flag is undocumented *
xprof
* This flag is undocumented *
python_targets_python3_11
* This flag is undocumented *
python_targets_python3_12
* This flag is undocumented *
python_targets_python3_13
* This flag is undocumented *
python_targets_python3_14
* This flag is undocumented *

