-
Notifications
You must be signed in to change notification settings - Fork 3.2k
Pull requests: jax-ml/jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[Mosaic] Support small second minor tiling for 16 bit reshape.
#31981
opened Sep 20, 2025 by
copybara-service
bot
Loading…
Fix a typo in pipelining pseudocode
kokoro:force-run
pull ready
Ready for copybara import and testing
#31979
opened Sep 20, 2025 by
IvyZX
Loading…
Remove dbias constraints in cudnn
nn.dot_product_attention.
#31961
opened Sep 19, 2025 by
acecchini
Loading…
[Mosaic] Increase the number of partial accumulators used in multi_reduction to expose more ILP.
#31947
opened Sep 19, 2025 by
copybara-service
bot
Loading…
[ROCm] Extend GPU handle pool isolation to hipSparse and replace macros with inline functions
#31937
opened Sep 18, 2025 by
AratiGanesh
Loading…
Docs: Add instructions for building jaxlib with AddressSanitizer (ASan)
#31914
opened Sep 18, 2025 by
Aniketsy
Loading…
[docs] Add 'Ref' to the documentation reference list
documentation
#31909
opened Sep 18, 2025 by
vfdev-5
Loading…
[Mosaic TPU] Add a new MultiDimReductionOp in the tpu dialect, which is almost the same as vector::MultiDimReductionOp except a new added boolean attribute
consistent_reduce_order. The attribute will be used to decide whether optimized reduction implementation is enabled.
#31905
opened Sep 18, 2025 by
copybara-service
bot
Loading…
Replace RBE Docker container image: use Docker image without pre-installed CUDA packages.
#31900
opened Sep 18, 2025 by
copybara-service
bot
Loading…
Update some stale CUDA 12 references in the documentation.
pull ready
Ready for copybara import and testing
#31896
opened Sep 18, 2025 by
hawkinsp
Loading…
[Mosaic:TPU] Have VectorLayout::join handle fully replicated layouts and fix bug in infer-vector-layout elementwise rule
#31891
opened Sep 17, 2025 by
copybara-service
bot
Loading…
Adds support for carried scalar and vector values in plsc.parallel_loop
#31873
opened Sep 17, 2025 by
copybara-service
bot
Loading…
[Mosaic:TPU][apply-vector-layout] tpu.concatenate for offsets outside of first tile
#31865
opened Sep 17, 2025 by
copybara-service
bot
Loading…
Mark args with jax.buffer_donor if output with same size exists
#31829
opened Sep 16, 2025 by
copybara-service
bot
Loading…
[Pallas] Improve error msg for out-of-bounds stores of blocks of outputs.
pull ready
Ready for copybara import and testing
#31825
opened Sep 15, 2025 by
jburnim
Loading…
Bump tj-actions/changed-files from 46.0.5 to 47.0.0
dependencies
Pull requests that update a dependency file
github_actions
Pull requests that update Github_actions code
#31821
opened Sep 15, 2025 by
dependabot
bot
Loading…
Bump hypothesis from 6.138.2 to 6.138.16
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#31820
opened Sep 15, 2025 by
dependabot
bot
Loading…
[Mosaic GPU][NFC] Infer the
is_signed property from the element type.
#31814
opened Sep 15, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.