Search Criteria
Package Details: python-jax 0.4.31-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.38 |
First Submitted: | 2019-08-04 14:29 (UTC) |
Last Updated: | 2024-08-13 00:25 (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-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
« First ‹ Previous 1 2 3
carlosal1015 commented on 2022-09-18 19:19 (UTC)
I could suggest using jax from conda-forge with ana/mini/conda over virtual environment.
bred commented on 2022-09-18 18:29 (UTC) (edited on 2022-09-18 18:35 (UTC) by bred)
The compilation stops with this error:
carlosal1015 commented on 2022-09-02 01:16 (UTC) (edited on 2022-09-02 01:17 (UTC) by carlosal1015)
I see that
python-etils
andpython-typing_extensions
are dependency according to this file.carlosal1015 commented on 2022-09-01 01:13 (UTC) (edited on 2022-09-01 04:01 (UTC) by carlosal1015)
I have an error message when try to install
python-zipp
orpython-typing_extensions
.I am not sure what could be a nice solution, maybe mark as conflict package.
steventrouble commented on 2021-03-09 00:53 (UTC) (edited on 2021-03-09 00:55 (UTC) by steventrouble)
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:
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 15:04 (UTC)
Well, the issue is binary compatibility between
tensorflow
andjax
sincejaxlib
heavily use code located intensorflow
source tree. Now I have figured out that these distributions are independent enough or there are at lease guarantees of independence (but there are still some risks) so we can buildjaxlib
from sources.I don't think that jaxlib is intended to use. I guess that the reason is that
jaxlib
is a private thin bridge between XLA client attensorflow
source tree andjax
. This bridge also contains XLA extensions like PRNG. Nevertheless,jaxlib
andjax
like seemingly should be different packages since they have different release cycles and say some critical hotfixes could come tojax
but notjaxlib
. So, I'll addpython-jaxlib
AUR package in near future.edh commented on 2021-01-12 13:46 (UTC) (edited on 2021-01-12 13:46 (UTC) by edh)
The packages are designed to be two separate entities. PyPi tells you so (two distinct setup.py files), the developers treat them as such and jaxlib can in theory be used independently from jax. On a very basic level, jaxlib simply wraps XLA code while jax is a very fancy numpy wrapper.
If you want, I can take over maintainership of the package if you no longer actively use it.
P.S. the package is still hopelessly outdated. Why did you unflag it?
daskol commented on 2021-01-12 12:19 (UTC) (edited on 2021-01-12 12:19 (UTC) by daskol)
Sure. I have made the package the long time ago when build from sources were much more tricky.
Why?
edh commented on 2021-01-05 14:21 (UTC)
Can you please split the package into jaxlib and jax. Furthermore, it would be awesome if you would build the package from source the way it is described in the installation section at https://jax.readthedocs.io/en/latest/developer.html#building-from-source .
« First ‹ Previous 1 2 3