Development is on Github: https://github.com/rocm-arch/python-jax-rocm Please open issues and PRs there instead of commenting.
Search Criteria
Package Details: python-jax-rocm 0.3.16-2
Package Actions
Git Clone URL: | https://aur.archlinux.org/python-jax-rocm.git (read-only, click to copy) |
---|---|
Package Base: | python-jax-rocm |
Description: | Differentiate, compile, and transform Numpy code (with ROCM) |
Upstream URL: | https://github.com/google/jax |
Licenses: | Apache |
Conflicts: | python-jax, python-jaxlib |
Provides: | python-jax, python-jaxlib |
Submitter: | wuxxin |
Maintainer: | wuxxin (acxz) |
Last Packager: | wuxxin |
Votes: | 0 |
Popularity: | 0.000000 |
First Submitted: | 2022-08-16 12:39 (UTC) |
Last Updated: | 2022-08-16 15:36 (UTC) |
Dependencies (17)
- absl-py (python-absl)
- miopen (miopen-opencl, opencl-amd-dev, miopen-hip)
- python (python36, python32, python37, python39, python38, python310, python312)
- python-etils
- python-numpy (python-numpy1.22, python-numpy-flame, python-numpy-mkl-bin, python-numpy-git, python-numpy-openblas, python-numpy-mkl)
- python-opt_einsum
- python-scipy (python-scipy-mkl-bin, python-scipy-git, python-scipy-mkl)
- python-six
- rccl (opencl-amd-dev)
- rocm-hip-runtime (opencl-amd)
- bazel (bazel-git, bazel024-bin, bazel024, bazel2, bazel3, bazel3-bin, bazelisk, bazelisk-bin, bazel5, bazelisk-git) (make)
- git (git-vfs, git-run-command-patch-git, git-git, git-fc) (make)
- miopen (miopen-opencl, opencl-amd-dev, miopen-hip) (make)
- python-pip (make)
- python-wheel (make)
- rccl (opencl-amd-dev) (make)
- rocm-hip-sdk (opencl-amd-dev) (make)
Required by (28)
- nanobind (requires python-jax) (check)
- python-aesara (requires python-jax) (optional)
- python-chex (requires python-jax)
- python-deepxde (requires python-jax) (optional)
- python-dm-haiku (requires python-jax)
- python-eagerpy-git (requires python-jax) (optional)
- python-equinox (requires python-jax)
- python-etils (requires python-jax) (check)
- python-etils (requires python-jax) (optional)
- python-flax (requires python-jax)
- python-galpy (requires python-jax) (optional)
- python-jax (requires python-jaxlib)
- python-jax-jumpy (requires python-jax) (optional)
- python-jax-jumpy (requires python-jaxlib) (optional)
- python-jaxtyping (requires python-jax) (optional)
- python-jmp (requires python-jax)
- python-optax (requires python-jax)
- python-orbax (requires python-jax)
- python-orbax-checkpoint (requires python-jax)
- Show 9 more...
Pinned Comments
wuxxin commented on 2022-08-16 12:40 (UTC)
Development is on Github: https://github.com/rocm-arch/python-jax-rocm Please open issues and PRs there instead of commenting.