Skip to content

Pull requests: jax-ml/jax

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

[Pallas] Add pallas distributed computation tutorial pull ready Ready for copybara import and testing
#22887 by justinjfu was merged Aug 9, 2024 Loading…
@sharadmv
90
Add an experimental JAX scaleDotProductAttention API to use XLA cuDNN fusedAttention feature. pull ready Ready for copybara import and testing
#18814 by Cjkkkk was merged Feb 13, 2024 Loading…
@superbobry @sharadmv
83
Additional input validation for transformations pull ready Ready for copybara import and testing
#10603 by JeppeKlitgaard was merged May 19, 2022 Loading…
2
@hawkinsp
59
simplify jax.numpy.tile pull ready Ready for copybara import and testing
#10228 by YouJiacheng was closed Jul 7, 2022 Loading…
@froystig
51
DOC: Improved docs for polynomial arithmetic functions of jax.numpy pull ready Ready for copybara import and testing
#21981 by rajasekharporeddy was merged Jul 2, 2024 Loading…
@dfm @jakevdp
46
adds scipy.stats.mode-esque function
#11475 by big-c-note was closed Oct 11, 2022 Loading…
@jakevdp
41
Improve performance of array integer indexing pull ready Ready for copybara import and testing
#10393 by lgeiger was merged May 6, 2022 Loading…
@jakevdp
41
[Waiting Reviews]: Fixes the parameters descriptions cla: yes
#1434 by joaogui1 was merged Oct 30, 2019 Loading…
37
Implemented mode function
#9141 by Yash621 was closed Oct 11, 2022 Loading…
@jakevdp
37
Support axis_index_groups in allreduce collectives cla: yes
#2382 by jekbradbury was merged May 8, 2020 Loading…
36
Implement complex logaddexp and logaddexp2 cla: yes pull ready Ready for copybara import and testing
#7141 by wdphy16 was merged Jul 7, 2021 Loading…
@jakevdp
35
WIP: Implementation of Scipy Bootstrap
#10871 by riven314 was closed Nov 3, 2023 Loading…
@jakevdp
33
When caching is enabled, also enable XLA caching features as well pull ready Ready for copybara import and testing
#22899 by trevor-m was merged Nov 13, 2024 Loading…
@dfm
33
[WIP] Implemented an extract_image_patches op using lax ops. cla: yes
#1154 by gehring was closed Oct 12, 2020 Loading…
31
Add b and return_sign functionality to scipy.special.logsumexp cla: yes
#3488 by dpfau was merged Jun 23, 2020 Loading…
@jakevdp
31
DOC: Improve docs of jnp.var and std pull ready Ready for copybara import and testing
#22167 by rajasekharporeddy was merged Jul 2, 2024 Loading…
@dfm @jakevdp
31
add scipy.special.zeta, scipy.special.polygamma cla: yes
#3385 by fehiepsi was merged Jun 16, 2020 Loading…
1
30
Add basic support for partitions within JAX. cla: yes
#2177 by chr1sj0nes was closed Mar 4, 2023 Loading…
27
implement np.convolve method in lax_numpy cla: yes
#1831 by michiboo was closed Apr 11, 2020 Loading…
27
Implemented numpy setxor1d cla: no
#2943 by joschkabraun was closed Jul 14, 2020 Loading…
@jakevdp
25
hilbert transform
#15121 by joglekara was closed Mar 24, 2023 Loading…
@jakevdp
24
[Pallas TPU] Add matrix multiplication tutorial pull ready Ready for copybara import and testing
#20021 by sharadmv was merged Jul 17, 2024 Loading…
24
replace int with operator.index for lax
#10398 by YouJiacheng was closed Nov 2, 2023 Loading…
@jakevdp
23
Added jax.random.vonmises
#13773 by harryjulian was closed Nov 6, 2023 Loading…
@jakevdp
22
Adding broadcast_argnums to pmap for allowing similar behaviour t… cla: yes
#1786 by botev was merged Feb 14, 2020 Loading…
21
ProTip! Follow long discussions with comments:>50.