Package Details: python-jax 0.2.9-1

Git Clone URL: https://aur.archlinux.org/python-jax.git (read-only, click to copy)
Package Base: python-jax
Description: Differentiate, compile, and transform Numpy code.
Upstream URL: https://github.com/google/jax/
Keywords: autograd deep-learning dl jit machine-learning ml
Licenses: Apache
Submitter: daskol
Maintainer: daskol (ei-grad)
Last Packager: daskol
Votes: 3
Popularity: 0.30
First Submitted: 2019-08-04 14:29
Last Updated: 2021-01-31 10:45

Latest Comments

daskol commented on 2021-02-01 15:04

The packages are designed to be two separate entities.

Well, the issue is binary compatibility between tensorflow and jax since jaxlib heavily use code located in tensorflow 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 build jaxlib from sources.

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.

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 at tensorflow source tree and jax. This bridge also contains XLA extensions like PRNG. Nevertheless, jaxlib and jax like seemingly should be different packages since they have different release cycles and say some critical hotfixes could come to jax but not jaxlib. So, I'll add python-jaxlib AUR package in near future.

edh commented on 2021-01-12 13:46

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

Furthermore, it would be awesome if you would build the package from source the way it is described in the installation section at ...

Sure. I have made the package the long time ago when build from sources were much more tricky.

Can you please split the package into jaxlib and jax.

Why?

edh commented on 2021-01-05 14:21

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 .