-
Notifications
You must be signed in to change notification settings - Fork 3.2k
Pull requests: jax-ml/jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Lower lu_pivots_to_permutation using a custom CUDA kernel.
cla: no
#6314
by ahoenselaar
was closed Apr 3, 2021
Loading…
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…
[shapecheck] Improve the shape computations to work better with the n…
cla: yes
#5286
by gnecula
was closed Jul 19, 2021
Loading…
[WIP] Implemented an
extract_image_patches op using lax ops.
cla: yes
#1154
by gehring
was closed Oct 12, 2020
Loading…
[Waiting Reviews]: Fixes the parameters descriptions
cla: yes
#1434
by joaogui1
was merged Oct 30, 2019
Loading…
WIP: Add a pylintrc to make it easire to use linter
cla: yes
#1442
by joaogui1
was merged Oct 5, 2019
Loading…
Adding
broadcast_argnums to pmap for allowing similar behaviour t…
cla: yes
#1786
by botev
was merged Feb 14, 2020
Loading…
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…
Proper error handling for scalar overflow when converted to 32-bit dtypes
cla: yes
#2019
by TuanNguyen27
was closed Apr 8, 2021
Loading…
Support axis_index_groups in allreduce collectives
cla: yes
#2382
by jekbradbury
was merged May 8, 2020
Loading…
Add basic support for partitions within JAX.
cla: yes
#2177
by chr1sj0nes
was closed Mar 4, 2023
Loading…
implement np.convolve method in lax_numpy
cla: yes
#1831
by michiboo
was closed Apr 11, 2020
Loading…