Skip to content

jax

JAX: Autograd and XLA

Available in

Overlay	Newest	Ebuilds	Last activity
blshkv-overlay GitHub ↗	0.4.25	1	24 h	details ›
pypi GitHub ↗	0.9.0	2	5 mo	details ›
pypi-sci GitHub ↗	0.2.26	3	6 mo	details ›

Versions & arches

Version	Overlay	amd64	arm64	x86	Committed
0.9.0	pypi	amd64 testing	arm64 untested	x86 testing	—	view · download · history ↗
0.8.3	pypi	amd64 testing	arm64 untested	x86 testing	—	view · download · history ↗
0.4.25 ^≈	blshkv-overlay	amd64 testing	arm64 testing	x86 testing	8 mo	view · download · history ↗
0.2.26 ^≈	pypi-sci	amd64 testing	arm64 untested	x86 untested	2 yr	view · download · history ↗
0.2.12 ^≈	pypi-sci	amd64 testing	arm64 untested	x86 untested	2 yr	view · download · history ↗
0.2.9 ^≈	pypi-sci	amd64 testing	arm64 untested	x86 untested	2 yr	view · download · history ↗

Use flags of 0.9.0

ci
cuda Enable NVIDIA CUDA support (computation on GPU)
cuda12
cuda12-local
cuda13
cuda13-local
k8s
minimum-jaxlib
rocm
tpu
xprof

4 expansion flags (python targets, ABIs, cpu flags…)

python_targets_python3_11
python_targets_python3_12
python_targets_python3_13
python_targets_python3_14

Runtime dependencies of 0.9.0

show 74 lines

dev-python/jaxlib[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

>=dev-python/ml-dtypes-0.5.0[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

>=dev-python/numpy-2.0[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

dev-python/opt-einsum[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

>=dev-python/scipy-1.13[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

minimum-jaxlib? (

dev-python/jaxlib[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

ci? (

dev-python/jaxlib[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

tpu? (

dev-python/jaxlib[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

tpu? (

dev-python/libtpu[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

tpu? (

dev-python/requests[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

cuda? (

dev-python/jaxlib[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

cuda? (

dev-python/jax-cuda12-plugin[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

cuda12? (

dev-python/jaxlib[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

cuda12? (

dev-python/jax-cuda12-plugin[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

cuda13? (

dev-python/jaxlib[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

cuda13? (

dev-python/jax-cuda13-plugin[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

cuda12-local? (

dev-python/jaxlib[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

cuda12-local? (

dev-python/jax-cuda12-plugin[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

cuda13-local? (

dev-python/jaxlib[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

cuda13-local? (

dev-python/jax-cuda13-plugin[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

rocm? (

dev-python/jaxlib[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

rocm? (

dev-python/jax-rocm7-plugin[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

k8s? (

dev-python/kubernetes[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

xprof? (

dev-python/xprof[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]

)

python_targets_python3_11? (

dev-lang/python:3.11

)

python_targets_python3_12? (

dev-lang/python:3.12

)

python_targets_python3_13? (

dev-lang/python:3.13

)

python_targets_python3_14? (

dev-lang/python:3.14

)