Pull Requests ROCm/jax - SemanticDiff

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