jax
GESVDJ support for ROCm GPUs in JAX
#580
Merged

GESVDJ support for ROCm GPUs in JAX #580

tsrw2048
tsrw2048 Added support for Gesvdj on ROCm devices. This commit also fixes usag…
936606fc
tsrw2048 tsrw2048 changed the base branch from rocm-jaxlib-v0.8.0 to rocm-jaxlib-v0.8.1 319 days ago
tsrw2048 tsrw2048 changed the base branch from rocm-jaxlib-v0.8.1 to rocm-jaxlib-v0.8.0 319 days ago
tsrw2048 tsrw2048 changed the base branch from rocm-jaxlib-v0.8.0 to rocm-jaxlib-v0.8.1 318 days ago
tsrw2048 tsrw2048 changed the base branch from rocm-jaxlib-v0.8.1 to rocm-jaxlib-v0.8.0 318 days ago
Ruturaj4
Ruturaj4 approved these changes on 2026-01-07
tsrw2048 tsrw2048 merged 9dbc4119 into rocm-jaxlib-v0.8.0 294 days ago
tsrw2048 tsrw2048 added open-upstream
tsrw2048 tsrw2048 added cherry-pick-candidate

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone