Skip to content

Pull requests: openxla/shardy

New pull request New
51 Open 1,212 Closed
51 Open 1,212 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[SDY:MPMD] Remove MIN_JAXLIB_VERSION_FOR_SCHEDULE_RULES
#1362 opened May 21, 2026 by copybara-service Bot Loading…
[SDY][PadForDivisibility] Add DotGeneralOpPattern to enforce kZero padding
#1359 opened May 20, 2026 by copybara-service Bot Loading…
@Varcho
Drop unused enable_late_inlining option from Shardy.
#1348 opened May 14, 2026 by copybara-service Bot Loading…
@ekayaaslan
[mpmd] Fail if inferred fragments remain.
#1344 opened May 14, 2026 by copybara-service Bot Loading…
Inline producer-consumer merging into UniquifyFunctionInputsOutputsPass.
#1339 opened May 13, 2026 by copybara-service Bot Loading…
[mpmd] Fix visibility of overridden methods.
#1336 opened May 12, 2026 by copybara-service Bot Loading…
Add missing loads for Bazel 8+ support
#1334 opened May 12, 2026 by cerisier Loading…
2
[XLA] Support xla.debug.Log in sharding passes
#1333 opened May 11, 2026 by copybara-service Bot Loading…
[mpmd] Extract WrapBlockArgReturnsPass from UniquifyFunctionInputsOutputsPass.
#1302 opened May 5, 2026 by copybara-service Bot Loading…
[ReplicaGroupV3][JAX+stablehlo][6/n] Emit RGV3 from JAX in shard_map
#1287 opened May 1, 2026 by copybara-service Bot Loading…
@Varcho
[mpmd] UniquifyAndMergeReturnsPass
#1274 opened Apr 30, 2026 by copybara-service Bot Loading…
Pass tokens unmodified through sdy-basic-propagate.
#1259 opened Apr 30, 2026 by copybara-service Bot Loading…
[mpmd] Add UniquifyAndMergeReturnsV2Pass: create-then-merge approach.
#1258 opened Apr 29, 2026 by copybara-service Bot Loading…
Add UniquifyAndMergeReturnsPass to merge return copies into existing fragments.
#1256 opened Apr 29, 2026 by copybara-service Bot Loading…
[SDY] Update JAX export and PartIR to provide a target SDY version
#1219 opened Apr 23, 2026 by copybara-service Bot Loading…
[SDY] Update IFRT HloProgram and IFRT IR Programs to use a target SDY version
#1218 opened Apr 23, 2026 by copybara-service Bot Loading…
[SDY] Update PjRt MLIR/HLO conversion API to accept an optional SDY target version during serialization
#1217 opened Apr 23, 2026 by copybara-service Bot Loading…
[mpmd] Merge sideways in export pipeline.
#1198 opened Apr 16, 2026 by copybara-service Bot Loading…
Avoid usage of deprecated jax.core APIs.
#1189 opened Apr 15, 2026 by copybara-service Bot Loading…
[consts] Add lots of logging
#1156 opened Apr 10, 2026 by copybara-service Bot Loading…
Remove mesh deduplication in LiftInlinedMeshesPass.
#1138 opened Apr 8, 2026 by copybara-service Bot Loading…
[consts] Adapt MPMD code to the new JAX handling of closed-over constants
#1115 opened Apr 7, 2026 by copybara-service Bot Loading…
Place uniquified return fragments adjacent to their producer
#1088 opened Mar 26, 2026 by copybara-service Bot Loading…
Fix Gather/Scatter sharding rule for dimensions not in start_index_map.
#1085 opened Mar 25, 2026 by copybara-service Bot Loading…
@ZixuanJiang
[mpmd] Add flag-guarded sideways merge of inferred fragments in export pipeline for forward only programs.
#1082 opened Mar 25, 2026 by copybara-service Bot Loading…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.