Search Criteria
Package Details: python-jax 0.4.30-1
Package Actions
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/google/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: | 7 |
Popularity: | 0.92 |
First Submitted: | 2019-08-04 14:29 (UTC) |
Last Updated: | 2024-06-19 07:06 (UTC) |
Dependencies (12)
- python-absl
- python-jaxlibAUR (python-jaxlib-binAUR, python-jaxlib-cudaAUR, python-jax-rocmAUR)
- python-ml-dtypes
- python-numpy (python-numpy-flameAUR, python-numpy-mkl-binAUR, python-numpy-gitAUR, python-numpy1AUR, python-numpy-mklAUR)
- python-opt_einsum
- python-scipy (python-scipy-mkl-binAUR, python-scipy-gitAUR, python-scipy-mklAUR)
- python-build (make)
- python-installer (python-installer-gitAUR) (make)
- python-setuptools (make)
- python-wheel (make)
- python-protobuf (python-protobuf-gitAUR) (optional) – Australis
- python-requests (optional) – Distributed computations
Required by (34)
- mitsuba3-git
- python-aesara (optional)
- python-array-api-compat (optional)
- python-autoray (check)
- python-chex
- python-d2l (optional)
- python-dm-haiku
- python-eagerpy-git (optional)
- python-equinox
- python-etils (optional)
- python-finitediffx
- python-flax
- python-galpy (optional)
- python-jax-jumpy (check)
- python-jax-jumpy (optional)
- python-jaxtyping (optional)
- python-jmp
- python-kernex
- python-mpi4jax
- python-optax
- Show 14 more...
Latest Comments
1 2 3 Next › Last »
daskol commented on 2024-06-19 07:39 (UTC)
@vnegens Python package
jax
does not requirebazel
whilejaxlib
requires indeed andbazel
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.
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 leavepython-jaxlib
for package without compilation and createpython-jaxlib-opt
package that requires compilation (and do the same forpython-jaxlib-cuda
).Romanovzky commented on 2024-02-21 10:16 (UTC)
It seems that this is failing to build with both
python-jaxlib-cuda
andpython-jaxlib
during the bazel build step (version7.0.2-1
) with errorERROR: error loading package '@@com_github_grpc_grpc//':
. Upon googling a bit, it seems that this is a bazel 7 regression. Downgrading to bazel6.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()
andbuild()
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 inbuild()
instead ofprepare()
.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 forpython-jax
). Now it is fixed.1 2 3 Next › Last »