jax
6259cfe3
- Fold transposes feeding the RHS of a matmul into a change in the dot dimension numbers.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
242 days ago
Fold transposes feeding the RHS of a matmul into a change in the dot dimension numbers. PiperOrigin-RevId: 755935066
References
#477 - CI: 06/20/25 upstream sync
#31381 - Remove old ROCm build code
#31720 - Fix ann_test.py numerical bug in target reshape
#31768 - [ROCm] Support lowering through PJRT_Triton_Extension
#32115 - Relax version requirements for ROCm Jax Plugin wheels
#33157 - Resolve undefined behavior in bitshift unit test
#33186 - Make nvidia version data optional for ROCm builds
#579 - Create rocm-test-requirements.txt
#580 - GESVDJ support for ROCm GPUs in JAX
#581 - Fix/pallas tests shared memory
#584 - Use plain bazel to test jax, use hermetic rocm dependency
#585 - update a test for checking zero ROCm GPU event
#34135 - [ROCm] update to test if there are GPU events when doing profiling on…
Author
jaswanth-
Committer
Google-ML-Automation
Parents
a4adf32f
Loading