Go
Home
Pricing
FAQ
Install
Home
Pricing
FAQ
Install
Login
via GitHub
ROCm/jax
Pull Requests
Commits
Open
Closed
port rocm/jax 0.9.2 cherrypicks to amd-main branch (#740)
#741 opened 2026-03-23 22:21 by
magaonka-amd
Fix: Scaled Matmul at rocm 9.0
#739 opened 2026-03-17 23:21 by
shurale-nkn
Fix: Scaled Matmul at rocm 9.1
#738 opened 2026-03-17 23:21 by
shurale-nkn
Add ROCm gesdd SVD backend, export back-compat data, and relaxed GPU …
enhancement
#734 opened 2026-03-06 16:41 by
cj401-amd
Add ROCm gesdd SVD backend, export back-compat data, and relaxed GPU …
enhancement
#733 opened 2026-03-06 16:36 by
cj401-amd
update test
#732 opened 2026-03-05 23:13 by
kiran-thumma
update test
#731 opened 2026-03-05 23:11 by
kiran-thumma
Revert "Fix HIP memory leaks in RNN kernels"
#730 opened 2026-03-05 17:43 by
magaonka-amd
Remove input_tensor_desc destruction in RNN kernels
cherry-pick-candidate
#729 opened 2026-03-05 17:13 by
magaonka-amd
Fix HIP memory leaks in RNN kernels
#728 opened 2026-03-05 16:25 by
magaonka-amd
Fix HIP memory leaks in RNN kernels
#727 opened 2026-03-05 16:20 by
magaonka-amd
Skip test_pmap on ROCm due to IndivisibleError with new pmap SPMD tiling
#725 opened 2026-03-04 20:47 by
AratiGanesh
[Testing do not merge] Add ROCm wheel build and test pipeline to continuous CI
#723 opened 2026-03-04 14:02 by
alekstheod
Fix batched Cholesky OOM on ROCm by bypassing hipSolver's external hipMalloc
#717 opened 2026-02-26 10:19 by
FlemingH
Improve ROCm pytest results handling
#716 opened 2026-02-25 17:13 by
psanal35
Implement Mosaic GPU detection and Auto-Skips
open-upstream
#714 opened 2026-02-21 12:22 by
gulsumgudukbay
fix poor performance of slogdet when input dimension n=1 or 2
enhancement
cherry-pick-candidate
#710 opened 2026-02-10 22:03 by
cj401-amd
Skip test_tridiagonal_solve_grad test 0.8.0
open-upstream
cherry-pick-candidate
#701 opened 2026-02-09 21:13 by
AratiGanesh
Skip testEighTinyNorm test 0.9.0
open-upstream
cherry-pick-candidate
#699 opened 2026-02-09 20:30 by
AratiGanesh
Enable scaled matmul tests on ROCm
#678 opened 2026-01-30 20:49 by
magaonka-amd
Rocm jaxlib v0.9.0
#671 opened 2026-01-28 23:56 by
Ruturaj4
Test pytest abort plugin
#670 opened 2026-01-28 23:32 by
gulsumgudukbay
update with ROCm profiling tests with kernel_details
#664 opened 2026-01-28 14:34 by
cj401-amd
remove skipTest associated to hipsparse transpose
#661 opened 2026-01-27 22:37 by
kliegeois
Support wheels as build deps
#638 opened 2026-01-22 16:41 by
alekstheod
Add script for running bazel tests on ROCm
#632 opened 2026-01-21 19:12 by
charleshofer
Use xla from rocm-jaxlib-v0.8.0 release branch
#599 opened 2026-01-14 14:44 by
alekstheod
Add CI workflow for Pytest ROCm
#594 opened 2026-01-12 18:17 by
mminutoli
Upstream ci download jax rocm wheels
#592 opened 2026-01-10 00:57 by
mminutoli
Adding run_pytest_rocm.sh
#591 opened 2026-01-09 02:35 by
gulsumgudukbay
Older