jax
ecaa24e6 - Fix `triu` and `tril` with explicit mode

Commit

5 days ago

Fix `triu` and `tril` with explicit mode PiperOrigin-RevId: 879388809

References

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

#35663 - Fix `triu` and `tril` with explicit mode

#35741 - [ROCm] Add individual skips to Mosaic tests

#35785 - [ROCm] Fix and simplify jax rocm plugin init script

#35821 - [ROCm] Use rocm plugin and pjrt wheels for pytest job in continuous wheel build pipeline

Author

yashk2810

yashk2810

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading