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

Lower lu_pivots_to_permutation using a custom CUDA kernel. cla: no
#6314 by ahoenselaar was closed Apr 3, 2021 Loading…
1
7 🚀 2
ahead-of-time lowering and compilation for jit cla: yes pull ready Ready for copybara import and testing
#7997 by froystig was merged Oct 8, 2021 Loading…
@froystig
17 🚀 2
[shapecheck] Improve the shape computations to work better with the n… cla: yes
#5286 by gnecula was closed Jul 19, 2021 Loading…
@gnecula
1 🚀 1
Update linalg.py cla: yes
#6541 by cdfreeman-google was closed May 25, 2021 Loading…
11 🚀 1
[WIP] Implemented an extract_image_patches op using lax ops. cla: yes
#1154 by gehring was closed Oct 12, 2020 Loading…
31
[Waiting Reviews]: Fixes the parameters descriptions cla: yes
#1434 by joaogui1 was merged Oct 30, 2019 Loading…
37
WIP: Add a pylintrc to make it easire to use linter cla: yes
#1442 by joaogui1 was merged Oct 5, 2019 Loading…
8
Implementation of np.ldexp and np.frexp cla: yes
#1529 by WindQAQ was merged Apr 1, 2020 Loading…
12
Adding broadcast_argnums to pmap for allowing similar behaviour t… cla: yes
#1786 by botev was merged Feb 14, 2020 Loading…
21
Fixed the documentation of api.linearize cla: yes
#2467 by gnecula was closed Mar 24, 2020 Loading…
6
Add ValueError to jax.numpy.squeeze function. cla: yes
#2307 by us was closed Apr 28, 2020 Loading…
8
Raise an error in np.var when the input array is complex and dtype is not cla: yes
#2288 by yurodiviy was merged May 5, 2020 Loading…
1
10
Proper error handling for scalar overflow when converted to 32-bit dtypes cla: yes
#2019 by TuanNguyen27 was closed Apr 8, 2021 Loading…
11
Support axis_index_groups in allreduce collectives cla: yes
#2382 by jekbradbury was merged May 8, 2020 Loading…
36
Implement nan-argmin/argmax cla: yes
#2398 by yurodiviy was merged Apr 28, 2020 Loading…
8
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
Bump jaxlib version cla: yes
#2201 by chr1sj0nes was closed Feb 10, 2020 Loading…
2
Combine backend and device arguments cla: yes
#1898 by skye was closed Mar 4, 2023 Loading…
9
Add a summary explaining the usage and context for JAX PRNG design. cla: yes
#2525 by jblespiau was merged May 26, 2020 Loading…
1
10
Add np.polyadd numpy function cla: yes
#3261 by ashridh was merged Jun 3, 2020 Loading…
16
Improve dtype test coverage for random_test cla: yes
#3254 by jakevdp was merged Jun 18, 2020 Loading…
6
Add np.polysub numpy function cla: yes
#3319 by ashridh was merged Jun 5, 2020 Loading…
6
Add segment_max operator. cla: yes
#3338 by alexdavies was closed May 12, 2021 Loading…
14
add scipy.special.zeta, scipy.special.polygamma cla: yes
#3385 by fehiepsi was merged Jun 16, 2020 Loading…
1
30
ProTip! Adding no:label will show everything without a label.