Default Branch

rocm-main 1f93b4b9b5 · Add Python and ROCm version matrix for rocm-main (#314) · Updated 2025-04-02 16:49:03 +00:00

Branches

ci-upstream-sync-164_1 aa06e1650f · Merge pull request #27687 from froystig:out-shard-bits · Updated 2025-04-03 00:44:28 +00:00	138 412
fix_hash_5.0_0 c4a1f4c454 · Update latest xla hash for JAX branch 0.5.0 · Updated 2025-04-02 23:31:34 +00:00	1319 25
ci-upstream-sync-163_1 f139192201 · Add OOB checks to jax.numpy array indexing · Updated 2025-04-02 02:17:57 +00:00	138 369
ci-upstream-sync-162_1 006a6a63fe · [Easy] Make pallas mesh grid handling more resilient to tuple names. · Updated 2025-04-01 05:02:29 +00:00	138 344
add-gfx1101-0.4.35 35e820cb96 · Add gfx1101 to build targets · Updated 2025-03-31 18:39:58 +00:00	2726 39
0.5.0-fix_build_clang19_tmp 85e80e426b · Fix C++23 build errors (#257) (#318) · Updated 2025-03-31 12:12:09 +00:00	1319 24
ci-upstream-sync-161_1 0edd715e96 · [mgpu/pallas] Expose WGMMA_TRANSPOSED layout · Updated 2025-03-30 23:12:33 +00:00	138 321
add-script-for-xla-hash-update 2776f2c5d8 · Add script for updating the XLA hash in workspace.bzl · Updated 2025-03-28 22:57:23 +00:00	1 1
find-pyversion-at-build-time-0.5.0 8c1eb0a2ef · Add canonicalization of python versions · Updated 2025-03-28 21:55:50 +00:00	1319 26
find-pyversion-at-build-time 5bf08753d6 · Add canonicalization of python versions · Updated 2025-03-28 19:23:54 +00:00	1 3
ci-upstream-sync-160_1 c5aa86a41a · Remove redundant filtering in the paged flash attention kernel · Updated 2025-03-28 05:26:11 +00:00	138 269
revert-squash 555d7ef8a1 · Revert "Merge pull request #297 from ROCm/prep-for-v0.5.1" · Updated 2025-03-27 17:01:25 +00:00	519 9
cherry_picks_from_5.0_to_5.1 7cd30d20e4 · Fix C++23 build errors (#257) (#318) · Updated 2025-03-27 15:42:47 +00:00	519 20
rocm-jaxlib-v0.5.1-work-fixes 6fba79468d · Fix C++23 build errors (#257) (#318) · Updated 2025-03-27 15:32:54 +00:00	519 20
ci-upstream-sync-159_1 8f25337a9f · [ragged-paged-attn] Combine k_pages and v_pages into kv_pages and zip on num_kv_heads. · Updated 2025-03-27 01:33:34 +00:00	138 235
rocm-jaxlib-v0.5.0-fix_perms_update d3b4701ced · address feedback · Updated 2025-03-26 19:28:24 +00:00	1319 23
rocm-jaxlib-v0.5.0-fix_perms ed5b01bd26 · fix permissions issue when running in patchelf · Updated 2025-03-26 17:25:13 +00:00	1319 22
ci-upstream-sync-158_1 89faa209e2 · Merge pull request #27017 from mattjj:input-saved-vjp · Updated 2025-03-26 05:03:56 +00:00	138 193
ci-upstream-sync-157_1 c8ccd7570a · Add functionality that let us do a "jax" only release · Updated 2025-03-25 18:48:29 +00:00	138 173
fix-auditwheel 4f9571eb2b · Fix auditwheel · Updated 2025-03-25 14:49:21 +00:00	138 150