jax
4d108a60 - Merge pull request #36537 from jakevdp:version-support

Commit

2 days ago

Merge pull request #36537 from jakevdp:version-support PiperOrigin-RevId: 895408651

References

#36522 - [ROCm] Added fixes to Bazel ROCm CI to use proper wheels.

#36537 - Update NumPy/SciPy version support

#747 - aiter mha kernels (ASM+CK) integration

#36572 - [ROCm] LSTM fix MIOpen wights layout

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading