Package Details: python-jax-opt-cuda-git

Git Clone URL: (read-only, click to copy)
Package Base: python-jax-opt-cuda-git
Description: Differentiate, compile, and transform Numpy code.
Upstream URL:
Keywords: autograd deep-learning dl jit machine-learning ml
Licenses: Apache
Conflicts: python-jax
Provides: python-jax
Submitter: daskol
Maintainer: daskol (ei-grad)
Last Packager: daskol
Votes: 2
Popularity: 0.000135
First Submitted: 2019-08-08 10:39
Last Updated: 2021-02-01 11:27

Latest Comments

steventrouble commented on 2021-03-09 00:54

Researchers: If you're having trouble getting CUDA to work, just make your own PKGBUILD that uses pip rather than trying to debug this or build from source. Trust me, it'll be faster and safer.

To get it to work, you can download this PKGBUILD and modify the "pip install" line to run the pip install line from the jax github page:

package() {
    pip install \
        --ignore-requires-python \
        --root=$pkgdir \
        "jax" \
        "jaxlib==0.1.61+cuda112" \

In general, this works for almost any pip library that isn't available in the AUR or the arch repos.

Happy jaxing!

daskol commented on 2021-02-01 11:30

The actual issue that jax stopped to build jaxlib extension directly in source tree and moved to building wheel. So, we should install the wheel instead of running manually.

AlexParkhomenko commented on 2021-01-30 14:33

yay -S python-jax --builddir /p
    To install the newly-built jaxlib wheel, run:
      pip install /p/python-jax-opt-cuda-git/src/python-jax-opt-cuda-git/dist/jaxlib-0.1.60-cp39-none-manylinux2010_x86_64.whl
    ==> Starting package()...
    python: can't open file '/p/python-jax-opt-cuda-git/src/python-jax-opt-cuda-git/build/': [Errno 2] No such file or directory
    ==> ERROR: A failure occurred in package().

maybe unzip ?

foxcub commented on 2020-05-01 19:39

This is missing a dependency: python-opt_einsum