PEP 503 repository index for jax[cuda]
Daily automatically rebuilds
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.
