archful = false
archive_name = "jax-0.7.1.tar.gz"
automode = true
extras = [
    "ci",
    "cpu",
    "cuda",
    "cuda12",
    "cuda12-local",
    "cuda13",
    "cuda13-local",
    "k8s",
    "minimum-jaxlib",
    "rocm",
    "tpu",
    "xprof",
]
license = "Apache-2.0"
license_files_present = true
name = "jax"
python_name = "python-jax"
source = "PyPI"
summary = "Differentiate, compile, and transform Numpy code."
url = "https://github.com/jax-ml/jax"
version = "0.7.1"