-
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
[shard-map] relax vmap-of-shmap error with spmd_axis_name
pull ready
Ready for copybara import and testing
[RFC] Add solver diagnostics for cg and bicgstab
cla: yes
#7825
opened Sep 6, 2021 by
PhilipVinc
•
Draft
Bump importlib-resources from 6.4.0 to 6.4.5
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#24018
opened Sep 30, 2024 by
dependabot
bot
Loading…
Bump kiwisolver from 1.4.5 to 1.4.7
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#24016
opened Sep 30, 2024 by
dependabot
bot
Loading…
Bump hypothesis from 6.102.4 to 6.122.3
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#25354
opened Dec 9, 2024 by
dependabot
bot
Loading…
Bump importlib-resources from 6.4.0 to 6.5.2
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#28352
opened Apr 28, 2025 by
dependabot
bot
Loading…
Bump fsspec from 2024.5.0 to 2024.10.0
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#24429
opened Oct 21, 2024 by
dependabot
bot
Loading…
Bump kiwisolver from 1.4.5 to 1.4.8
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#28350
opened Apr 28, 2025 by
dependabot
bot
Loading…
Bump fsspec from 2024.5.0 to 2025.5.1
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#29011
opened May 26, 2025 by
dependabot
bot
Loading…
Bump fonttools from 4.51.0 to 4.55.3
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#25512
opened Dec 16, 2024 by
dependabot
bot
Loading…
Bump the pip group across 1 directory with 2 updates
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#28835
opened May 19, 2025 by
dependabot
bot
Loading…
Bump fonttools from 4.59.2 to 4.60.0
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#32012
opened Sep 22, 2025 by
dependabot
bot
Loading…
Enhance TPU Matmul RHS transpose fusion to handle intervening slices.
#31625
opened Sep 8, 2025 by
copybara-service
bot
Loading…
Canonicalize to vclamps (for vllm fp8 quant, maybe other uses)
#31452
opened Aug 31, 2025 by
copybara-service
bot
Loading…
docs: Add note about computation perturbation in debug.print/breakpoint
#26682
opened Feb 22, 2025 by
alhridoy
Loading…
[pallas:mgpu] warp specialized pipeline with special delay_release.
#31483
opened Sep 2, 2025 by
copybara-service
bot
Loading…
Consolidate material on debugging NaNs.
kokoro:force-run
pull ready
Ready for copybara import and testing
#24989
opened Nov 19, 2024 by
emilyfertig
Loading…
Add scalar_first parameter to jax.scipy.spatial.transform.Rotation.from_quat
#30826
opened Aug 7, 2025 by
kamalraj23
Loading…
Bump hypothesis from 6.138.16 to 6.140.2
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#32188
opened Sep 29, 2025 by
dependabot
bot
Loading…
[Pallas:MGPU] Add a primitive to reduce TCGEN05 to WGMMA for interleaved 0:16 data input
#31460
opened Sep 1, 2025 by
copybara-service
bot
Loading…
remove the annoying closed_call in scan lowering
kokoro:force-run
pull ready
Ready for copybara import and testing
#31254
opened Aug 22, 2025 by
mattjj
Loading…
Put gpu_ondevice_tracing py extension into jax/_src/lib
#31239
opened Aug 21, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Mix and match filters to narrow down what you’re looking for.