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

[docs] New tutorial on Tracing
#30703 opened Aug 1, 2025 by melissawm Loading… updated Sep 10, 2025
@emilyfertig
13
feat(scipy.special): add erfcx — scaled complementary error function
#29455 opened Jun 13, 2025 by KAVYANSHTYAGI Loading… updated Sep 5, 2025
@jakevdp
9
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… updated Aug 7, 2025
6
[mosaic-gpu] add multicast ptr support to TMA with overlapped gemm and all reduce examples
#28679 opened May 12, 2025 by Amir-19 Loading… updated May 28, 2025
@apaszke
13
replaced np.array with np.fromfile for performance improvements
#27018 opened Mar 9, 2025 by ParamThakkar123 Loading… updated Mar 10, 2025
@jakevdp
1
Retry connection errors to combat flakiness in google metadata server during tpu initialization (reopens #24432)
#26919 opened Mar 4, 2025 by dlwh Loading… updated Mar 7, 2025
2
Bessel Functions to Match scipy
#17038 opened Aug 9, 2023 by benjaminpope Loading… updated Nov 17, 2024
@jakevdp
51
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… updated Nov 16, 2024
@jakevdp
42
Refactor & Document
#21691 opened Jun 6, 2024 by Arkay92 Loading… updated Jun 6, 2024
2
Add comb function to jax._src.scipy.special
#18389 opened Nov 4, 2023 by Phionx Loading… updated Jan 23, 2024
1
@jakevdp
24
Add docstring to PositionalSharding
#17327 opened Aug 28, 2023 by rajveer43 Loading… updated Nov 9, 2023
1
@jakevdp
31
Update JAX README - add shard_map information (pmap deprecation)
#17210 opened Aug 21, 2023 by 8bitmp3 Loading… updated Aug 22, 2023
2
Adding a new transposed convolution function to lax cla: yes
#5772 opened Feb 18, 2021 by yang-song Loading… updated Jun 21, 2023
@hawkinsp
13
fix Dropout's bug
#16229 opened Jun 2, 2023 by fmscole Loading… updated Jun 10, 2023
@froystig
5
Extend Common-Gotchas docs: speed of jnp.array
#11054 opened Jun 10, 2022 by onnoeberhard Loading… updated Oct 4, 2022
1
4
Add NumPy backend cla: yes
#3923 opened Jul 31, 2020 by juliuskunze Loading… updated Jun 17, 2021
@mattjj
54
ProTip! Add no:assignee to see everything that’s not assigned.