Package Details: python-jax 0.4.35-1

Git Clone URL: https://aur.archlinux.org/python-jax.git (read-only, click to copy)
Package Base: python-jax
Description: Composable transformations of Python+NumPy programs
Upstream URL: https://github.com/jax-ml/jax/
Keywords: autograd deep-learning jax jit machine-learning ml xla
Licenses: Apache
Groups: jax
Submitter: daskol
Maintainer: daskol (ei-grad)
Last Packager: daskol
Votes: 10
Popularity: 1.38
First Submitted: 2019-08-04 14:29 (UTC)
Last Updated: 2024-10-23 00:02 (UTC)

Latest Comments

1 2 3 4 Next › Last »

risyasin commented on 2024-10-20 18:10 (UTC) (edited on 2024-10-20 18:10 (UTC) by risyasin)

Hey in case you need the latest version 0.4.34 as of 20th October. Here, you can find it: https://git.evrima.net/dev/python-jax/-/commit/2e473e4d6a3878cc9d5b66ca73a29ea25e577603 I've already contacted with Daniel, I am sure, he will update soon.

daskol commented on 2024-09-26 21:24 (UTC)

@truncs Yes, indeed. I noticed that. At the moment, it is tricky to build python-jaxlib* due to multiple issues (i.e. clang18 release, yanked jax v0.4.32 release, and build configuration of XLA for CUDA target). It takes some time. Downgrade to jax v0.4.31, please. I'll add dependency constraint on python-jaxlib in order to prevent automatic update.

truncs commented on 2024-09-26 21:18 (UTC)

@daskol jax and jaxlib are no longer tagged as separate releases. Please see the discussion over here https://github.com/jax-ml/jax/discussions/23952#discussioncomment-10768018

daskol commented on 2024-06-19 07:39 (UTC)

@vnegens Python package jax does not require bazel while jaxlib requires indeed and bazel is in requirements there.

vnegens commented on 2024-06-02 10:42 (UTC)

Hello,

For those who don't know, the package is dependent from bazel. If bazel is not installed on your system, it won't build. Is it possible to put a dependency check at the beginning of the build ? Regards.

daskol commented on 2024-02-27 17:33 (UTC)

@Romanovzky I'll check and fix compilation with the new JAX release.

Also, on a different note, would there the possibility to install the package without compilation?

Yes, it's possible. However, there are some vendored dependencies. This reduces distribution size. If there are other people who would just pip install jax, we might leave python-jaxlib for package without compilation and create python-jaxlib-opt package that requires compilation (and do the same for python-jaxlib-cuda).

Romanovzky commented on 2024-02-21 10:16 (UTC)

It seems that this is failing to build with both python-jaxlib-cuda and python-jaxlib during the bazel build step (version 7.0.2-1) with error ERROR: error loading package '@@com_github_grpc_grpc//':. Upon googling a bit, it seems that this is a bazel 7 regression. Downgrading to bazel 6.4.0.-1 solved the problem. Also, on a different note, would there the possibility to install the package without compilation? I.e. "a la" pip install?

Henry-ZHR commented on 2023-10-21 04:24 (UTC)

At least it's necessary for Yay users. Yay runs prepare() and build() in different shells.

daskol commented on 2023-10-21 00:04 (UTC)

@Henry-ZHR It is not necessary (see python-tensorflow as an example) but I agree that shorter is better.

zxeoc commented on 2023-10-11 04:05 (UTC)

PKGBUILD asks for "jax-$pkgver-.whl" but there is only jax-$pkgver.whl(no "-" after $pkgver)