jax
3dacc957
- Rollback "Custom batching rule for Cholesky that ensures symmetrize."
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
193 days ago
Rollback "Custom batching rule for Cholesky that ensures symmetrize." Reverts ebba464b591fe2a8a4bdf6440672a78b471b22d2 PiperOrigin-RevId: 775356791
References
#29710 - Rollback "Custom batching rule for Cholesky that ensures symmetrize."
#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
btaba
Committer
Google-ML-Automation
Parents
69796d34
Loading