Skip to content

Pull requests: jax-ml/jax

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

Adding the elliptical integral functions ellipj, ellipk, ellipe. cla: yes
#2492 opened Mar 23, 2020 by botev Loading…
@mattjj
18
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
feat(pallas): Optimize Pallas Attention + Benchmark
#17328 opened Aug 28, 2023 by jon-chuang Loading…
@sharadmv
17
[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
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…
[ROCm] Extend GPU handle pool isolation to hipSparse and replace macros with inline functions
#31937 opened Sep 18, 2025 by AratiGanesh Loading…
8
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…
1
8
[pallas:mgpu] warp specialized pipeline with special delay_release.
#31483 opened Sep 2, 2025 by copybara-service bot Loading…
Clarify wait_recv semantics
#30000 opened Jul 7, 2025 by tengyifei Loading…
3
Adds DeserializeExecutable Overloads for the new FfiHostCallback code path.
#31469 opened Sep 1, 2025 by yliu120 Loading…
@emilyfertig
1
[Mosaic GPU] Add support for multimem ops on WG strided arrays
#32248 opened Oct 1, 2025 by copybara-service bot Loading…
@apaszke
Support For Roofline Analysis Of Pallas Kernels
#32245 opened Oct 1, 2025 by copybara-service bot Loading…
[Mosaic GPU] Improve checks on the WG strided transfers
#32247 opened Oct 1, 2025 by copybara-service bot Loading…
@apaszke
Conditional scan JEP
#13184 opened Nov 10, 2022 by patrick-kidger Loading…
@sharadmv
6
ProTip! Exclude everything labeled bug with -label:bug.