jax
Enable test_nvfp4_gradient_clip tests on ROCm
#620

Merged

Enable test_nvfp4_gradient_clip tests on ROCm #620

gulsumgudukbay merged 1 commit into rocm-jaxlib-v0.8.0 from phambinh/enable-nvfp4-gradient-clip-tests-rocm

Enable test_nvfp4_gradient_clip tests on ROCm

bdeef72f

phambinhfin requested a review 231 days ago

gulsumgudukbay approved these changes on 2026-01-20

phambinhfin added open-upstream

phambinhfin added cherry-pick-candidate

gulsumgudukbay merged 21f71a9f into rocm-jaxlib-v0.8.0 231 days ago

gulsumgudukbay deleted the phambinh/enable-nvfp4-gradient-clip-tests-rocm branch 231 days ago

Reviewers

gulsumgudukbay

Assignees

No one assigned

Labels

open-upstream cherry-pick-candidate

Milestone

No milestone