![]() * py-jaxlib: add spack-built ROCm support * fix style * py-jaxlib 0.4.38 rocm support * py-jaxlib 0.4.38 rocm support * add comgr dependency * changes for ROCm external and enable till 0.4.38 * enable version of py-jax * add jax+rocm to ci * add conflict for cuda and remove py-jaxlib from aarch64 pipeline * Update var/spack/repos/builtin/packages/py-jaxlib/package.py Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com> * add conflict for aarch64 --------- Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com> |
||
---|---|---|
.. | ||
spack |