Home » Oreps » Jax » Pull Requests
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Watch
1
Commit Activity
Pull Requests
Directory Browser
Velocity History
Releases

Pull Requests

Review 1674 pull requests

Closed
[mosaic_gpu] `parse_indices` no longer allows `slice()` with an `ir.Value`

Data pending calculation for pull request

No reviewer comments found.
#40786 with 1 commit opened about 17 hours ago by copybara-service[bot]
test_982641428
Opened and closed September 18
Review Code View on GitHub
Total Diff Delta
0
Target
500
Team average
79
Team average reflects all PRs in the jax repo over the past 30 days.
0
Diff Delta
500 under target
Total comments
0
Target
10
Team average
0
Team average reflects all PRs in the jax repo over the past 30 days.
0
comments
10 under target
Merged
[Roll forward -- Take 2] Optimize large JAX Mesh construction

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#40785 with 1 commit opened about 17 hours ago by copybara-service[bot]
test_983759493
Opened and closed September 18
Review Code View on GitHub
Total Diff Delta
0
Target
500
Team average
79
Team average reflects all PRs in the jax repo over the past 30 days.
0
Diff Delta
500 under target
Total comments
0
Target
10
Team average
0
Team average reflects all PRs in the jax repo over the past 30 days.
0
comments
10 under target
Open: XS
[ROCm] Skip the TheRock pre-release CI legs on release runs
Opened by Marco Minutoli about 23 hours ago
Aug 30
No reviewer comments found yet.
#40784 with 0 commits opened about 23 hours ago by mminutoli
user/mminutoli/disable-rocm-latest-on-release
Next action:
Review code
View at GitHub
2 pull request requested reviewers:
Avatar of charleshofer
Avatar of nitins17
Last action:
No commits within processing range found
Open: XS
[pallas] Preserve out_avals memory spaces in `_mpmd_map_discharge_rule`
Opened by copybara-service[bot] 1 day ago
Aug 30
No reviewer comments found yet.
#40783 with 0 commits opened 1 day ago by copybara-service[bot]
test_983558282
Next action:
Code review
Review code
View at GitHub
0 pull request requested reviewers:
No reviewers nor requested reviewers found.
Last action:
No commits within processing range found
Open: XS
Reverts bf979cc0fdd9c7418b03a47ddce38dd62f7131bb
Opened by copybara-service[bot] 1 day ago
Aug 30
No reviewer comments found yet.
#40782 with 0 commits opened 1 day ago by copybara-service[bot]
test_983474769
Next action:
Code review
Review code
View at GitHub
0 pull request requested reviewers:
No reviewers nor requested reviewers found.
Last action:
No commits within processing range found
Open: XS
[Mosaic TPU] Support more convolution cases.
Opened by copybara-service[bot] 1 day ago
Aug 30
No reviewer comments found yet.
#40781 with 0 commits opened 1 day ago by copybara-service[bot]
test_983521138
Next action:
Code review
Review code
View at GitHub
0 pull request requested reviewers:
No reviewers nor requested reviewers found.
Last action:
No commits within processing range found
Open: XS
[Pallas:SC] Migrate compress store lowering to tpu.vector_compress_store when layout passes are on
Opened by copybara-service[bot] 1 day ago
Aug 30
No reviewer comments found yet.
#40780 with 0 commits opened 1 day ago by copybara-service[bot]
test_982093153
Next action:
Code review
Review code
View at GitHub
0 pull request requested reviewers:
No reviewers nor requested reviewers found.
Last action:
No commits within processing range found
Open: XS
[Pallas:SC] Lower swaps to tpu.compress_store_vreg or tpu.store without layout passes
Opened by copybara-service[bot] 1 day ago
Aug 30
No reviewer comments found yet.
#40779 with 0 commits opened 1 day ago by copybara-service[bot]
test_983470154
Next action:
Code review
Review code
View at GitHub
0 pull request requested reviewers:
No reviewers nor requested reviewers found.
Last action:
No commits within processing range found
Open: XS
[Pallas:SC][NFC] Make compress swaps explicit
Opened by copybara-service[bot] 1 day ago
Aug 30
No reviewer comments found yet.
#40778 with 0 commits opened 1 day ago by copybara-service[bot]
test_983470155
Next action:
Code review
Review code
View at GitHub
0 pull request requested reviewers:
No reviewers nor requested reviewers found.
Last action:
No commits within processing range found
Open: XS
[Pallas][MPMD] Preserve out_avals memory spaces in _mpmd_map_discharge_rule
Opened by copybara-service[bot] 1 day ago
Aug 30
No reviewer comments found yet.
#40777 with 0 commits opened 1 day ago by copybara-service[bot]
test_983480985
Next action:
Code review
Review code
View at GitHub
0 pull request requested reviewers:
No reviewers nor requested reviewers found.
Last action:
No commits within processing range found
Merged
Lower shard_map in partial explicit manual mode correctly by using the aval.sharding.spec instead of `in_specs` since the former contain more info (explicit mesh axes too) while in_specs only refer to axes passed in axis_names.

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#40776 with 1 commit opened 1 day ago by copybara-service[bot]
test_983407126
Opened and closed September 17
Review Code View on GitHub
Total Diff Delta
151
While under review
151
Target
500
Team average
79
Team average reflects all PRs in the jax repo over the past 30 days.
151
Diff Delta
349 under target
Total comments
0
Target
10
Team average
0
Team average reflects all PRs in the jax repo over the past 30 days.
0
comments
10 under target
Merged
[Pallas] Skip test_async_sc_tc_prefetch_vmem with libtpu < 0.0.49.

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#40775 with 1 commit opened 1 day ago by copybara-service[bot]
test_983459171
Opened and closed September 17
Review Code View on GitHub
Total Diff Delta
6
While under review
6
Target
500
Team average
79
Team average reflects all PRs in the jax repo over the past 30 days.
6
Diff Delta
494 under target
Total comments
0
Target
10
Team average
0
Team average reflects all PRs in the jax repo over the past 30 days.
0
comments
10 under target
Open: XS
[fuser] Fuser fixes
Opened by copybara-service[bot] 1 day ago
Aug 30
No reviewer comments found yet.
#40774 with 0 commits opened 1 day ago by copybara-service[bot]
test_982718121
Next action:
Code review
Review code
View at GitHub
0 pull request requested reviewers:
No reviewers nor requested reviewers found.
Last action:
No commits within processing range found
Open: XS
Update XLA dependency to use revision http://github.com/openxla/xla/commit/0f4b29b0e7de1bc9e4d0bd7c8a626ca02bb7833c.
Opened by copybara-service[bot] 1 day ago
Aug 30
No reviewer comments found yet.
#40773 with 0 commits opened 1 day ago by copybara-service[bot]
test_983443987
Next action:
Code review
Review code
View at GitHub
0 pull request requested reviewers:
No reviewers nor requested reviewers found.
Last action:
No commits within processing range found
Merged
Fix a CI test failure on Windows.

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#40772 with 1 commit opened 1 day ago by copybara-service[bot]
test_983442027
Opened and closed September 17
Review Code View on GitHub
Total Diff Delta
3
While under review
3
Target
500
Team average
79
Team average reflects all PRs in the jax repo over the past 30 days.
3
Diff Delta
498 under target
Total comments
0
Target
10
Team average
0
Team average reflects all PRs in the jax repo over the past 30 days.
0
comments
10 under target
GitClear uses cookies to ensure you get the best experience on our website. Learn more about our cookie policy