Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/jorenham/jax_pep503

PEP 503 repository index for jax[cuda]
https://github.com/jorenham/jax_pep503

dependency-manager jax package-manager packaging packaging-python pep503 pip pypi python-poetry python3

Last synced: about 2 months ago
JSON representation

PEP 503 repository index for jax[cuda]

Host: GitHub
URL: https://github.com/jorenham/jax_pep503
Owner: jorenham
License: apache-2.0
Created: 2022-01-12T01:02:50.000Z (almost 3 years ago)
Default Branch: master
Last Pushed: 2024-05-01T00:30:22.000Z (8 months ago)
Last Synced: 2024-05-01T15:54:12.812Z (8 months ago)
Topics: dependency-manager, jax, package-manager, packaging, packaging-python, pep503, pip, pypi, python-poetry, python3
Language: Python
Homepage: https://jorenham.github.io/jax_pep503/
Size: 1.18 MB
Stars: 21
Watchers: 3
Forks: 4
Open Issues: 0
Metadata Files:
- Readme: README.md
- License: LICENSE

Awesome Lists containing this project

README

        # PEP 503 repository index for `jax[cuda]`

*Daily automatically rebuilds*

## Repository

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

## Example: Adding `jax[cuda]` to a [Poetry](https://python-poetry.org/) project

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

    ```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:

    ```bash

    poetry add jax[cuda]

    ```

See the [example project](example_project) for more details.

## See also

- [PEP 503 – Simple Repository API](https://peps.python.org/pep-0503/)

- [google/jax](https://github.com/google/jax)

- [google/jax#5410 – Please provide PEP 503 compliant indices for CUDA versions of packages](https://github.com/google/jax/issues/5410)

- [jax-releases](https://storage.googleapis.com/jax-releases/) (not compatible with PEP 503)