jax
4ac5e1ec - Merge pull request #33560 from jax-ml:ufunc-sigs

Commit

38 days ago

Merge pull request #33560 from jax-ml:ufunc-sigs PiperOrigin-RevId: 839465677

References

#33560 - Add exceptions for`numpy` `ufunc`s in `tests/lax_numpy_test.py::NumpySignaturesTest.testWrappedSignatureMatch`.

#34135 - [ROCm] update to test if there are GPU events when doing profiling on…

#592 - Upstream ci download jax rocm wheels

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading