Skip to content

Pull requests: AI-Hypercomputer/pathways-utils

New pull request New
9 Open 145 Closed
9 Open 145 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

Internal change
#154 opened Jan 8, 2026 by copybara-service bot Loading…
No public description
#153 opened Jan 7, 2026 by copybara-service bot Loading…
Require JAX>=0.7.0
#152 opened Jan 7, 2026 by copybara-service bot Loading…
Internal change
#150 opened Jan 6, 2026 by copybara-service bot Loading…
Updated jax.extend.ifrt_programs import to import the module and make an alias for the jax.extend.ifrt_programs.ifrt_programs variable.
#149 opened Jan 6, 2026 by copybara-service bot Loading…
Add find_intermediate_sharding to optimize resharding.
#146 opened Dec 19, 2025 by copybara-service bot Loading…
This change introduces reshard_with_intermediate_sharding which will first look for intermediate shardings, perform all intermediate resharding, and then perform the final reshard into the out sharding.
#145 opened Dec 19, 2025 by copybara-service bot Loading…
Update "Shared Pathways Service" README
#143 opened Dec 18, 2025 by guptaaka Loading…
5
Adding some support for ProfileOptions to jax.profiler.start_trace for Pathways.
#130 opened Dec 8, 2025 by copybara-service bot Loading…
ProTip! Updated in the last three days: updated:>2026-01-07.