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
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: | 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-openclAUR, opencl-amd-devAUR, miopen-hip)
- python (python32AUR, python37AUR, python310AUR)
- python-etilsAUR
- python-numpy (python-numpy-flameAUR, python-numpy-mkl-binAUR, python-numpy-gitAUR, python-numpy-mklAUR)
- python-opt_einsum
- python-scipy (python-scipy-mkl-binAUR, python-scipy-gitAUR, python-scipy-mklAUR)
- python-six
- rccl (opencl-amd-devAUR)
- rocm-hip-runtime (opencl-amdAUR)
- bazel (bazel-gitAUR, bazel024-binAUR, bazel3AUR, bazel3-binAUR, bazel5AUR, bazelisk-gitAUR, bazelisk-binAUR, bazeliskAUR) (make)
- git (git-gitAUR) (make)
- miopen (miopen-openclAUR, opencl-amd-devAUR, miopen-hip) (make)
- python-pip (make)
- python-wheel (make)
- rccl (opencl-amd-devAUR) (make)
- rocm-hip-sdk (opencl-amd-devAUR) (make)
Required by (38)
- mitsuba3-git (requires python-jax)
- nanobind (requires python-jax) (check)
- python-aesara (requires python-jax) (optional)
- python-chex (requires python-jax)
- python-d2l (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-finitediffx (requires python-jax)
- python-flax (requires python-jax)
- python-galpy (requires python-jax) (optional)
- python-jax (requires python-jaxlib)
- python-jax-jumpy (requires python-jax) (check)
- python-jax-jumpy (requires python-jax) (optional)
- python-jaxtyping (requires python-jax) (optional)
- python-jmp (requires python-jax)
- python-kernex (requires python-jax)
- python-mpi4jax (requires python-jax)
- Show 18 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.