jax
b2ffd47b - Merge pull request #33601 from jakevdp:dtype-protocol

Commit

44 days ago

Merge pull request #33601 from jakevdp:dtype-protocol PiperOrigin-RevId: 839460911

References

#31768 - [ROCm] Support lowering through PJRT_Triton_Extension

#33601 - Add support for NumPy 2.4 __numpy_dtype__ protocol

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

#34304 - [ROCm] Add ROCm platform support for psend/precv collective operations

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading