Search results

update a test for checking zero ROCm GPU event

cherry-pick-candidateMark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR.Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR.

open-upstreamTag when you want a copy of this PR to be opened on upstreamTag when you want a copy of this PR to be opened on upstream

Status: Open (in progress).

ROCm/jaxnumber 585

#585 In ROCm/jax;

· cj401-amd opened on Dec 18, 2025

[ROCm] Implement RNN support

open-upstreamTag when you want a copy of this PR to be opened on upstreamTag when you want a copy of this PR to be opened on upstream

Status: Open (in progress).

ROCm/jaxnumber 171

#171 In ROCm/jax;

· Ruturaj4 opened on Dec 2, 2024

Update http to https in amd artifactory url.

open-upstreamTag when you want a copy of this PR to be opened on upstreamTag when you want a copy of this PR to be opened on upstream

Status: Open (in progress).

ROCm/jaxnumber 164

#164 In ROCm/jax;

· Ruturaj4 opened on Nov 26, 2024

[ROCm] Add argument to override base docker in dockerfile

open-upstreamTag when you want a copy of this PR to be opened on upstreamTag when you want a copy of this PR to be opened on upstream

Status: Open (in progress).

ROCm/jaxnumber 163

#163 In ROCm/jax;

· Ruturaj4 opened on Nov 26, 2024

[ROCm] ci build and dockerfile changes

open-upstreamTag when you want a copy of this PR to be opened on upstreamTag when you want a copy of this PR to be opened on upstream

Status: Open (in progress).

ROCm/jaxnumber 161

#161 In ROCm/jax;

· Ruturaj4 opened on Nov 26, 2024

[ROCm] Change run_multi_gpu set opts

open-upstreamTag when you want a copy of this PR to be opened on upstreamTag when you want a copy of this PR to be opened on upstream

Status: Open (in progress).

ROCm/jaxnumber 156

#156 In ROCm/jax;

· Ruturaj4 opened on Nov 26, 2024