jorenham / jax_pep503 Goto Github PK

View Code? Open in Web Editor NEW

21.0 21.0 4.0 1.54 MB

PEP 503 repository index for jax[cuda]

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

License: Apache License 2.0

Python 94.52% HTML 5.48%

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

jax_pep503's People

Contributors

Stargazers

Watchers

Forkers

sourcery-ai-bot hatzav-ox neilgirdhar davecwright3

jax_pep503's Issues

CPU & GPU

I have access to two machines. One that has a gpu and another that doesn't.

Is it possible to set-up the pyproj.toml file so that the right version of jax is used for each machine?

jax[tpu]

Hi,

I have been trying to install jax[tpu] using poetry but I couldn't. I have been using this command to install the jax version that works with tpu using pip:

python -m pip install --upgrade "jax[tpu]==0.3.10" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html

However, I couldn't find a way to have it work with poetry. Do you have any ideas? Thanks!

Possible escaping issue in package links?

I'm not sure if this is the source of the problem, but it seems like I'm not able to install any of the package versions that have a "+" in the version. e.g. "0.4.4+cuda11.cudnn86"

I got here from this issue comment (python-poetry/poetry#6956 (comment)). It looks like the Mac version of the jaxlib hasn't been updated since 0.3.18 (https://jorenham.github.io/jax_pep503/jaxlib/). It would be wonderful to be able to use this!

Thanks!!

Recommend Projects