Search Criteria
Package Details: python-jax-rocm 0.7.1-1
Package Actions
Git Clone URL: | https://aur.archlinux.org/python-jax-rocm.git (read-only, click to copy) |
---|---|
Package Base: | python-jax-rocm |
Description: | XLA library for JAX (jaxlib for ROCM) |
Upstream URL: | https://github.com/jax-ml/jax |
Keywords: | python,jax,rocm,google,amd |
Licenses: | Apache-2.0 |
Groups: | jax |
Conflicts: | python-jaxlib |
Provides: | python-jaxlib |
Submitter: | wuxxin |
Maintainer: | carlosal1015 |
Last Packager: | carlosal1015 |
Votes: | 3 |
Popularity: | 0.038733 |
First Submitted: | 2022-08-16 12:39 (UTC) |
Last Updated: | 2025-09-28 22:25 (UTC) |
Dependencies (13)
- miopen-hip (opencl-amd-devAUR)
- python-absl
- python-flatbuffers (python-flatbuffers-gitAUR)
- python-ml-dtypes
- python-scipy (python-scipy-gitAUR, python-scipy-mkl-tbbAUR, python-scipy-mklAUR, python-scipy-mkl-binAUR)
- rccl (opencl-amd-devAUR)
- rocm-hip-runtime (opencl-amdAUR)
- bazel (bazel3-binAUR, bazelisk-gitAUR, bazelisk-binAUR, bazelisk) (make)
- python-build (make)
- python-installer (make)
- python-setuptools (make)
- python-wheel (make)
- rocm-hip-sdk (opencl-amd-devAUR) (make)
Required by (5)
- python-blackjax (requires python-jaxlib)
- python-jax (requires python-jaxlib)
- python-jax-bin (requires python-jaxlib)
- python-safetensors (requires python-jaxlib) (optional)
- python-safetensors-bin (requires python-jaxlib) (optional)
Latest 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.