gpo.zugaina.org

Search Portage & Overlays:

dev-python/jax

Differentiate, compile, and transform Numpy code

Screenshots

  • jax-0.11.1
    ~amd64
    rocm test python_targets_python3_13

    View      Download      Browse     License: Apache-2.0   
    Overlay: pypi-sci
  • jax-0.9.2
    ~amd64
    rocm test python_targets_python3_12 python_targets_python3_13

    View      Download      Browse     License: Apache-2.0   
    Overlay: pypi-sci
  • jax-0.9.0
    ~amd64 ~x86
    ci 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 ~x86
    ci 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 ~x86
    python_targets_python3_12 python_targets_python3_13 python_targets_python3_14

    View      Download      Browse     License:   
  • jax-0.4.14
    ~amd64
    test python_targets_python3_12 python_targets_python3_13

    View      Download      Browse     License: Apache-2.0   
    Overlay: pypi-sci
  • jax-0.2.26
    ~amd64
    test python_targets_python3_12 python_targets_python3_13

    View      Download      Browse     License: Apache-2.0   
    Overlay: pypi-sci
  • jax-0.2.12
    ~amd64
    test python_targets_python3_12 python_targets_python3_13

    View      Download      Browse     License: Apache-2.0   
    Overlay: pypi-sci
  • jax-0.2.9
    ~amd64
    test python_targets_python3_12 python_targets_python3_13

    View      Download      Browse     License: Apache-2.0   
    Overlay: pypi-sci

Runtime Dependencies

jax-0.11.1


~dev-python/jaxlib-0.11.1[python_targets_python3_13
      (-)
?
]

>=dev-python/ml-dtypes-0.5.0[python_targets_python3_13
      (-)
?
]

>=dev-python/numpy-2.1[python_targets_python3_13
      (-)
?
]

dev-python/opt-einsum[python_targets_python3_13
      (-)
?
]

>=dev-python/scipy-1.15[python_targets_python3_13
      (-)
?
]

rocm?
      ( ~dev-python/jax-rocm7-plugin-0.11.1[python_targets_python3_13
            (-)
      ?
] )

python_targets_python3_13?
      ( dev-lang/python:3.13 )

jax-0.9.2



~dev-python/jaxlib-0.9.2[python_targets_python3_12
      (-)
?
,python_targets_python3_13
      (-)
?
]


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


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


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


>=dev-python/scipy-1.14[python_targets_python3_12
      (-)
?
,python_targets_python3_13
      (-)
?
]

rocm?
      ( ~dev-python/jax-rocm7-plugin-0.9.2[python_targets_python3_12
            (-)
      ?
,python_targets_python3_13
            (-)
      ?
] )

python_targets_python3_12?
      ( dev-lang/python:3.12 )

python_targets_python3_13?
      ( dev-lang/python:3.13 )

jax-0.9.0





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 )

jax-0.8.3





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 )

jax-0.4.25


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 )

jax-0.4.14


python_targets_python3_12?
      ( dev-lang/python:3.12 )

python_targets_python3_13?
      ( dev-lang/python:3.13 )

jax-0.2.26


python_targets_python3_12?
      ( dev-lang/python:3.12 )

python_targets_python3_13?
      ( dev-lang/python:3.13 )

jax-0.2.12


python_targets_python3_12?
      ( dev-lang/python:3.12 )

python_targets_python3_13?
      ( dev-lang/python:3.13 )

jax-0.2.9


python_targets_python3_12?
      ( dev-lang/python:3.12 )

python_targets_python3_13?
      ( dev-lang/python:3.13 )