jax
Fix: Scaled Matmul at rocm 9.0
#739
Open
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Overview
Commits
1
Changes
View On
GitHub
Fix: Scaled Matmul at rocm 9.0
#739
shurale-nkn
wants to merge 1 commit into
rocm-jaxlib-v0.9.0
from
fix_scaled_matmul_dot_test_0.9.0
Add ROCm support for scaled matrix multiplication and scaled dot lowe…
0187ef92
Login to write a write a comment.
Login via GitHub
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub