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 1652 pull requests

Closed
[ci] pin zipp to fix documentation builds

Data pending calculation for pull request

No reviewer comments found.
#41870 with 1 commit opened 23 days ago by copybara-service[bot]
test_1004216073
Opened and closed October 21
Review Code View on GitHub
Total Diff Delta
0
Target
500
Team average
84
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
Reduce redundant digamma and lgamma evaluations in special functions.

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#41869 with 1 commit opened 23 days ago by copybara-service[bot]
test_1004201622
Open October 21 - October 22
Review Code View on GitHub
Total Diff Delta
122
While under review
122
Target
500
Team average
84
Team average reflects all PRs in the jax repo over the past 30 days.
122
Diff Delta
378 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
[xla] Never CSE buffer values; reject a buffer written via two operands.
Opened by copybara-service[bot] 23 days ago
Oct 24
No reviewer comments found yet.
#41868 with 1 commit opened 23 days ago by copybara-service[bot]
test_1003523485
Next action:
Code review
Review code
View at GitHub
0 pull request requested reviewers:
No reviewers nor requested reviewers found.
Last action:
Oct 24 by copybara-service[bot]
Merged
Improve accuracy of lax.digamma to within 1 ULP across all float types.

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#41867 with 1 commit opened 23 days ago by copybara-service[bot]
test_1000339049
Open October 21 - October 22
Review Code View on GitHub
Total Diff Delta
1,078
While under review
1,078
Target
500
Team average
84
Team average reflects all PRs in the jax repo over the past 30 days.
1,078
Diff Delta
578 over 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
Fall back to caller location for bare-NameLoc MLIR callsites in source_mapper.

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#41865 with 1 commit opened 23 days ago by copybara-service[bot]
test_1003520552
Opened and closed October 21
Review Code View on GitHub
Total Diff Delta
0
Target
500
Team average
84
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
[JAX] Fix jax.scipy.special.loggamma nan at zero (jax-ml/jax#38240)

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#41864 with 1 commit opened 23 days ago by copybara-service[bot]
test_1004112690
Open October 21 - October 22
Review Code View on GitHub
Total Diff Delta
0
Target
500
Team average
84
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
[Mosaic GPU] Guard `bf16` <-> `f8e4m3fn`/`f8e5m2` conversions on Blackwell+ (`sm_100`+).

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#41863 with 1 commit opened 23 days ago by copybara-service[bot]
test_1003994976
Opened and closed October 21
Review Code View on GitHub
Total Diff Delta
36
While under review
36
Target
500
Team average
84
Team average reflects all PRs in the jax repo over the past 30 days.
36
Diff Delta
464 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
[JAX] Fix jax.scipy.linalg.toeplitz (and hankel) NaN propagation

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#41860 with 1 commit opened 23 days ago by copybara-service[bot]
test_1004027934
Open October 21 - October 22
Review Code View on GitHub
Total Diff Delta
0
Target
500
Team average
84
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
[Pallas TPU] Add missing f-string prefixes in get/swap lowering error messages
 
 0.0 days Authoring commits before pull request opened for review
 
 0.3 days Awaiting first review
 
 0.1 days Revising work while processing 1 reviewer who left 1 comment

Total time for pull request 0.4 business days from first commit authored to merge

1 reviewer
awaited review for 2.5 business hours
#41858 with 1 commit opened 23 days ago by ayaka14732
pallas-sc-fstring
Opened and closed October 21
Review Code View on GitHub
Total Diff Delta
0
Target
500
Team average
84
Team average reflects all PRs in the jax repo over the past 30 days.
0
Diff Delta
500 under target
Total comments
1
Target
10
Team average
0
Team average reflects all PRs in the jax repo over the past 30 days.
1
comment
9 under target
Merged
[typing] Avoid non-contiguous overloads in splash_attention_kernel.py
 
 0.0 days Authoring commits before pull request opened for review
 
 0.1 days Awaiting first review
 
 0.1 days Revising work while processing 1 reviewer who left 1 comment

Total time for pull request 0.2 business days from first commit authored to merge

1 reviewer
awaited review for 0.5 business hours
#41856 with 1 commit opened 23 days ago by stroxler
fix-noncontiguous-overload
Opened and closed October 21
Review Code View on GitHub
Total Diff Delta
0
Target
500
Team average
84
Team average reflects all PRs in the jax repo over the past 30 days.
0
Diff Delta
500 under target
Total comments
1
Target
10
Team average
0
Team average reflects all PRs in the jax repo over the past 30 days.
1
comment
9 under target
Closed
Roll forward: lower cumulative reductions to chlo.ScanOp on TPU

Data pending calculation for pull request

No reviewer comments found.
#41855 with 1 commit opened 23 days ago by copybara-service[bot]
test_957659209
Open October 21 - October 23
Review Code View on GitHub
Total Diff Delta
0
Target
500
Team average
84
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
Closed
[Pallas:MGPU] Support transposing the swizzled dimension in `UnswizzleRef`.

Data pending calculation for pull request

No reviewer comments found.
#41854 with 1 commit opened 23 days ago by copybara-service[bot]
test_1003935315
Open October 21 - October 22
Review Code View on GitHub
Total Diff Delta
226
Before PR submitted
226
Target
500
Team average
84
Team average reflects all PRs in the jax repo over the past 30 days.
226
Diff Delta
274 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:fuser] Do not scalar prefetch inside nested calls in `push_block_spec`

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#41852 with 1 commit opened 23 days ago by copybara-service[bot]
test_1003941194
Open October 21 - October 22
Review Code View on GitHub
Total Diff Delta
99
While under review
99
Target
500
Team average
84
Team average reflects all PRs in the jax repo over the past 30 days.
99
Diff Delta
401 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
Version-gate TPU-specific `inline=False` lowering rule for `lgamma_p` in `jax._src.lax.special` on `libtpu < 0.0.52`.

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#41845 with 1 commit opened 23 days ago by copybara-service[bot]
test_1003852896
Open October 21 - October 23
Review Code View on GitHub
Total Diff Delta
15
While under review
15
Target
500
Team average
84
Team average reflects all PRs in the jax repo over the past 30 days.
15
Diff Delta
485 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
Add worst-case math library inputs to JAX numerics accuracy tests.

Data pending calculation for pull request

Pull request was merged without approval or review comments found.
#41842 with 2 commits opened 23 days ago by copybara-service[bot]
test_1003859875
Opened and closed October 21
Review Code View on GitHub
View Post-Merge Changes
Total Diff Delta
441
Post-merge
296
While under review
146
Target
500
Team average
84
Team average reflects all PRs in the jax repo over the past 30 days.
441
Diff Delta
59 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