[ROCm]: fix: JAX/TE sharding compatibility and tmem reduction foundations (PR1)#4191
Open
cj401-amd wants to merge 5 commits into
Open
[ROCm]: fix: JAX/TE sharding compatibility and tmem reduction foundations (PR1)#4191cj401-amd wants to merge 5 commits into
cj401-amd wants to merge 5 commits into
Google CLA / cla/google
succeeded
Jun 19, 2026 in 7s
✅ All contributors are covered under a CLA with Google
See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).
ℹ️ Googlers: Go here to view more details and manage scans for this pull request.
Details
The following contributors were found for this pull request:
✅ a779105 Author: @cj401-amd <ch****in@amd.com>
(Only the first commit for a unique contributor is listed.)
Loading