jax
46d36125 - Merge pull request #34573 from jakevdp:refactor-hijax-tests

Commit

4 days ago

Merge pull request #34573 from jakevdp:refactor-hijax-tests PiperOrigin-RevId: 859889757

References

#34573 - [hijax,test] refactor hijax test suite

#34598 - Implement approx_tanh for ROCm using OCML tanh function

#34600 - Enable scaled_matmul tests on ROCm

#34601 - Enable neural network tests on ROCm

#34602 - Fix and enable Pallas ops tests on ROCm

#34603 - Enable miscellaneous tests on ROCm

#34615 - [ROCm] Skip test_batch_axis_sharding_jvp on ROCm

#34644 - [ROCm] Added skip for "test_prim_tridiagonal_solve" tests on ROCm.

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading