Skip to content

/ jax

New pull request New
106 Open 2,436 Closed
106 Open 2,436 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels.
Projects
Filter by project
Milestones
Filter by milestone
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
rm padded env from masking.py
cla: yes
#4119 opened Aug 21, 2020 by j-towns
1
Support overriding implementations of JAX functions within a scope
cla: yes
#4117 opened Aug 21, 2020 by shoyer • Draft
2
make custom_jvp handle all tracers gracefully
cla: yes
#4112 opened Aug 20, 2020 by mattjj • Draft
Add support for negative axes to vmap.
cla: yes
#4111 opened Aug 20, 2020 by hawkinsp • Approved
7
Add a boolean to _check_shapelike to accept or reject shapes
cla: yes
#4108 opened Aug 20, 2020 by SIben • Approved
1
@gnecula
3
Simplify the interface for host_callback.id_tap
cla: yes
#4101 opened Aug 19, 2020 by shoyer
11
[jax2tf] Added conversion for scatter*_p primitives.
cla: yes
#4091 opened Aug 18, 2020 by SIben
@gnecula
9
Add support for binding axis_name in gmap
cla: yes
#4090 opened Aug 18, 2020 by apaszke • Approved
9
Upstream propagate interpreter from oryx to jax.experimental
cla: yes
#4088 opened Aug 17, 2020 by sharadmv
jaxlib build fixes.
cla: yes
#4066 opened Aug 14, 2020 by skye • Approved
Use local version identifiers to distribute cuda jaxlib wheels.
cla: yes
#4065 opened Aug 14, 2020 by skye • Approved
14
Support mask(pmap)
cla: yes
#4046 opened Aug 13, 2020 by j-towns • Draft
@mattjj
7
Grad of int
cla: yes
#4039 opened Aug 12, 2020 by LenaMartens • Draft 4 of 6
omnistaging on by default
cla: yes
#4038 opened Aug 12, 2020 by mattjj • Draft
5
Add early check for pmap inputs larger than number of local devices.
cla: yes
#4037 opened Aug 12, 2020 by skye • Approved
2
Fixing weird behavior in segment_sum when num_segments is None.
cla: yes
#4034 opened Aug 12, 2020 by alvarosg • Approved
6
[jax2tf] Clean up code for XlaGather, experimental_compile not necessary
cla: yes
#4030 opened Aug 12, 2020 by gnecula
1
Add GMRES to `jax.scipy.sparse.linalg`
cla: yes
#4025 opened Aug 11, 2020 by gehring • Draft 0 of 4
12
custom_jvp/vjp closure issues, just experimenting for now
cla: yes
#4008 opened Aug 10, 2020 by mattjj • Draft
Allow JAX objects to be represented by multiple buffers
cla: yes
#3983 opened Aug 6, 2020 by jakevdp 4 of 6
Add NumPy backend
cla: yes
#3923 opened Jul 31, 2020 by JuliusKunze • Changes requested
@mattjj
51
Clarify error message in scan transpose
cla: yes
#3917 opened Jul 30, 2020 by jekbradbury • Approved
2
Add test for dtype coverage of jax.numpy ufuncs
cla: yes
#3913 opened Jul 30, 2020 by jakevdp • Draft
WIP: exploring weakly-dtyped device arrays
cla: yes
#3903 opened Jul 29, 2020 by jakevdp • Draft
Make PyTreeDef pickleable for Multiprocessing
cla: yes
#3868 opened Jul 27, 2020 by bionicles
3
ProTip! no:milestone will show everything without a milestone.
You can’t perform that action at this time.