Metadata-Version: 2.1
Name: jax-rocm7-plugin
Version: 0.6.0
Summary: JAX Plugin for AMD GPUs (ROCm:unknown)
Home-page: https://github.com/jax-ml/jax
Author: Ruturaj4
Author-email: Ruturaj.Vaidya@amd.com
License: Apache-2.0
Classifier: Development Status :: 3 - Alpha
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.9
Description-Content-Type: text/markdown
Requires-Dist: jax-rocm7-pjrt ==0.6.0
Provides-Extra: with_rocm
Requires-Dist: amd-rocm-hip-runtime-devel-instinct ; extra == 'with_rocm'
Requires-Dist: amd-rocm-hip-runtime-instinct ; extra == 'with_rocm'
Requires-Dist: amd-hipblas-instinct ; extra == 'with_rocm'
Requires-Dist: amd-hipsparse-instinct ; extra == 'with_rocm'
Requires-Dist: amd-hipsolver-instinct ; extra == 'with_rocm'
Requires-Dist: amd-miopen-hip-instinct ; extra == 'with_rocm'
Requires-Dist: amd-rocm-llvm-instinct ; extra == 'with_rocm'
Requires-Dist: amd-rocm-language-runtime-instinct ; extra == 'with_rocm'
Requires-Dist: amd-rccl-instinct ; extra == 'with_rocm'
Requires-Dist: amd-hipfft-instinct ; extra == 'with_rocm'
Requires-Dist: amd-rocm-device-libs-instinct ; extra == 'with_rocm'
Requires-Dist: amd-hipsparselt-instinct ; extra == 'with_rocm'

