Skip to content

Pull requests: jax-ml/jax

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

Add comb function to jax._src.scipy.special
#18389 opened Nov 4, 2023 by Phionx Loading…
1
@jakevdp
24
Refactor & Document
#21691 opened Jun 6, 2024 by Arkay92 Loading…
2
Add NumPy backend cla: yes
#3923 opened Jul 31, 2020 by juliuskunze Loading…
@mattjj
54
Determine symmetric linear op fro CG from abstract output dtype pull ready Ready for copybara import and testing
#23486 opened Sep 6, 2024 by Joshuaalbert Loading…
@jakevdp
42
Add nd_loop and Enable block_n tiling for all_gather_lhs_matmul kokoro:force-run pull ready Ready for copybara import and testing
#29822 opened Jun 27, 2025 by hanzlfs Loading…
6
Add docstring to PositionalSharding
#17327 opened Aug 28, 2023 by rajveer43 Loading…
1
@jakevdp
31
Retry connection errors to combat flakiness in google metadata server during tpu initialization (reopens #24432)
#26919 opened Mar 4, 2025 by dlwh Loading…
2
Update JAX README - add shard_map information (pmap deprecation)
#17210 opened Aug 21, 2023 by 8bitmp3 Loading…
2
Adding a new transposed convolution function to lax cla: yes
#5772 opened Feb 18, 2021 by yang-song Loading…
@hawkinsp
13
feat(scipy.special): add erfcx — scaled complementary error function
#29455 opened Jun 13, 2025 by KAVYANSHTYAGI Loading…
@jakevdp
9
fix Dropout's bug
#16229 opened Jun 2, 2023 by fmscole Loading…
@froystig
5
Extend Common-Gotchas docs: speed of jnp.array
#11054 opened Jun 10, 2022 by onnoeberhard Loading…
1
4
[docs] New tutorial on Tracing
#30703 opened Aug 1, 2025 by melissawm Loading…
@emilyfertig
13
Bessel Functions to Match scipy
#17038 opened Aug 9, 2023 by benjaminpope Loading…
@jakevdp
51
[mosaic-gpu] add multicast ptr support to TMA with overlapped gemm and all reduce examples
#28679 opened May 12, 2025 by Amir-19 Loading…
@apaszke
13
replaced np.array with np.fromfile for performance improvements
#27018 opened Mar 9, 2025 by ParamThakkar123 Loading…
@jakevdp
1
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.