Search Criteria
Package Details: python-jax 0.4.35-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/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: | 9 |
Popularity: | 0.95 |
First Submitted: | 2019-08-04 14:29 (UTC) |
Last Updated: | 2024-10-23 00:02 (UTC) |
Dependencies (12)
- python-absl
- python-jaxlibAUR (python-jax-rocmAUR, python-jaxlib-cudaAUR, python-jaxlib-binAUR)
- python-ml-dtypes
- python-numpy (python-numpy-flameAUR, python-numpy-mkl-binAUR, python-numpy-gitAUR, python-numpy1AUR, python-numpy-mklAUR, python-numpy-mkl-tbbAUR)
- python-opt_einsum
- python-scipy (python-scipy-mkl-binAUR, python-scipy-gitAUR, python-scipy-mklAUR, python-scipy-mkl-tbbAUR)
- 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 (33)
- 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 13 more...
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 onpython-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 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)
1 2 3 4 Next › Last »