Package Details: python-jax 0.4.29-1

Git Clone URL: (read-only, click to copy)
Package Base: python-jax
Description: Composable transformations of Python+NumPy programs
Upstream URL:
Keywords: autograd deep-learning jax jit machine-learning ml xla
Licenses: Apache
Groups: jax
Submitter: daskol
Maintainer: daskol (ei-grad)
Last Packager: daskol
Votes: 6
Popularity: 0.35
First Submitted: 2019-08-04 14:29 (UTC)
Last Updated: 2024-06-11 07:22 (UTC)

Latest Comments

1 2 3 Next › Last »

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


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)

Henry-ZHR commented on 2023-10-07 04:43 (UTC) (edited on 2023-10-07 04:44 (UTC) by Henry-ZHR)

export JAX_RELEASE should be in build() instead of prepare().

Henry-ZHR commented on 2023-09-01 04:41 (UTC)

Failed to build: FileNotFoundError: [Errno 2] No such file or directory: 'jax-jax-v0.4.15/dist/jax-0.4.15-py3-*-*.whl'

There is only jax-0.4.15.dev20230901-py3-none-any.whl

daskol commented on 2023-07-18 21:17 (UTC)

@jb53 The issue was caused by python-ml-dtypes (dependency for python-jax). Now it is fixed.

daskol commented on 2023-07-17 08:59 (UTC)

@jb53 It is a bit odd since python-jax does not depend on pybind11 at all whilst python-jaxlib depends on it but pybind11 is vendored and is already resolved dependency.