jax
Add `device test` unit tests for ROCm (JAX v0.9.0)
#705

Merged

Add `device test` unit tests for ROCm (JAX v0.9.0) #705

tsrw2048 merged 1 commit into rocm-jaxlib-v0.9.0 from add-device-test-v0.9.0

Added device test for ROCm.

36ac633f

tsrw2048 requested a review 118 days ago

tsrw2048 added open-upstream

tsrw2048 added cherry-pick-candidate

tsrw2048 changed the title ~~Add device test unit tests for ROCm (JAX v0.9.0)~~ Add `device test` unit tests for ROCm (JAX v0.9.0) 118 days ago

magaonka-amd approved these changes on 2026-02-10

charleshofer approved these changes on 2026-02-10

tsrw2048 merged 0988e6da into rocm-jaxlib-v0.9.0 118 days ago

Reviewers

charleshofer

magaonka-amd

Assignees

No one assigned

Labels

open-upstream cherry-pick-candidate

Milestone

No milestone