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