jax
Backporting `device test` unit tests for ROCm from JAX v0.9.0 to v0.8.2.
#706

Merged

Backporting `device test` unit tests for ROCm from JAX v0.9.0 to v0.8.2. #706

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

Add device test unit tests for ROCm.

dd595e39

tsrw2048 requested a review 128 days ago

tsrw2048 added open-upstream

tsrw2048 added cherry-pick-candidate

gulsumgudukbay approved these changes on 2026-02-10

tsrw2048 merged 2890b29c into rocm-jaxlib-v0.8.2 122 days ago

Reviewers

gulsumgudukbay

Assignees

No one assigned

Labels

open-upstream cherry-pick-candidate

Milestone

No milestone