Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
452 Open 14,955 Closed
452 Open 14,955 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

RFC: add pytree-compatible dataclass decorator
#24664 opened Nov 1, 2024 by jakevdp • Draft
@jakevdp
32 🎉 8
RFC: specify jit static args via Static annotation enhancement New feature or request
#24705 opened Nov 4, 2024 by jakevdp • Draft
@jakevdp
11 🎉 3
Adding the elliptical integral functions ellipj, ellipk, ellipe. cla: yes
#2492 opened Mar 23, 2020 by botev Loading…
@mattjj
18 🎉 2
Support mask(pmap) cla: yes
#4046 opened Aug 13, 2020 by j-towns • Draft
@mattjj
7 🎉 1
Prefix PRNG Part 2: implement 1D prefix PRNG cla: yes
#2181 opened Feb 6, 2020 by j-towns Loading…
17 🎉 1
[CI] Ignore: Tune the B200 tests per accelerator
#32121 opened Sep 25, 2025 by MichaelHudgins • 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…
1
Add wheel build and py_import tests in JAX presubmit.
#31045 opened Aug 14, 2025 by copybara-service bot Loading…
feat(pallas): Optimize Pallas Attention + Benchmark
#17328 opened Aug 28, 2023 by jon-chuang Loading…
@sharadmv
17
Add support for device kwarg in astype, and add matching utility func
#21086 opened May 6, 2024 by Micky774 Loading…
28
[mutable-arrays] change more callsites to use discharge-state2 pull ready Ready for copybara import and testing
#31135 opened Aug 18, 2025 by mattjj Loading…
@mattjj
Adds docs on multi-controller JAX fault tolerance.
#32218 opened Sep 30, 2025 by mwhittaker • Draft
@mwhittaker
Adds DeserializeExecutable Overloads for the new FfiHostCallback code path.
#31469 opened Sep 1, 2025 by yliu120 Loading…
@emilyfertig
1
[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…
[Mosaic:TPU] Better tpu.memref_reshape support for packed types by improving the reshaped layout inference
#31418 opened Aug 28, 2025 by copybara-service bot Loading…
Fix Jax OSS CI CI Optional GPU Presubmit Label to flag PR to run additional GPU testing not in standard presubmits
#31562 opened Sep 4, 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…
ProTip! Exclude everything labeled bug with -label:bug.