Skip to content

Latest commit

 

History

History
42 lines (28 loc) · 1.05 KB

README.md

File metadata and controls

42 lines (28 loc) · 1.05 KB
Raw

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.

See also