PEP 503 repository index for jax[cuda]

Daily automatically rebuilds

Important

The CUDA wheels of JAX are now published on PyPI, so this index will no longer be needed: jax-ml/jax#5410 (comment)

Repository

https://jorenham.github.io/jax_pep503/

Example: Adding jax[cuda] to a Poetry project

• Add the repository as a secondary source to your pyproject.toml:

  ...
  
  [[tool.poetry.source]]
  name = "PyPI"
  priority = "primary"
  
  [[tool.poetry.source]]
  name = "jorenham/jax_pep503"
  url = "https://jorenham.github.io/jax_pep503/"
  priority = "supplemental"
  
  [tool.poetry.dependencies]
  ...

• Now you can add jax[cuda] using:

  poetry add jax[cuda]

See the example project for more details.

See also

• PEP 503 – Simple Repository API
• google/jax
• google/jax#5410 – Please provide PEP 503 compliant indices for CUDA versions of packages
• jax-releases (not compatible with PEP 503)