Skip to content

/ jax

New pull request New
110 Open 2,509 Closed
110 Open 2,509 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
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
Add comments for residuals from f_bwd.
cla: yes
#4244 opened Sep 10, 2020 by zhangqiaorjc
Revert "Revert "Delete batching.last. (#4148)""
cla: yes
#4242 opened Sep 10, 2020 by hawkinsp
Change lax_linalg.lu to return a permutation representation of the pa…
cla: yes
#4241 opened Sep 9, 2020 by hawkinsp
[jax2tf] Moved the limitations for XlaSort to correctness_stats
cla: yes
#4237 opened Sep 9, 2020 by gnecula
@gnecula
6
Add a test that fails due to unsupported strides.
cla: yes
#4230 opened Sep 8, 2020 by zhangqiaorjc • Approved
2
Expose jax.image.scale_and_translate
cla: yes
#4229 opened Sep 8, 2020 by johnpjf • Changes requested
1
Extend axis env while translating the pmapped jaxpr to XLA
cla: yes
#4227 opened Sep 8, 2020 by apaszke • Approved
1
implement jnp.apply_over_axes
cla: yes
#4225 opened Sep 8, 2020 by jakevdp
[jax2tf] Add tests for the conversion of conv_general_dilated (WIP).
cla: yes
#4222 opened Sep 8, 2020 by SIben • Draft
fix docs build by adding requirement
cla: yes
#4205 opened Sep 4, 2020 by froystig
1
Call check_arraylike() in jax.numpy reductions
cla: yes
#4195 opened Sep 2, 2020 by jakevdp • Approved
Gmres qr
cla: yes
#4175 opened Aug 28, 2020 by alewis • Draft
16
Allow vmapping all_to_all and implement a (slow) CPU and GPU translation
cla: yes
#4170 opened Aug 28, 2020 by apaszke • Approved
1
4
raise_to_shaped: preserve weak_type by default
cla: yes
#4161 opened Aug 27, 2020 by jakevdp • Approved
2
Support overriding implementations of JAX functions within a scope
cla: yes
#4117 opened Aug 21, 2020 by shoyer • Draft
@gnecula
5
make custom_jvp handle all tracers gracefully
cla: yes
#4112 opened Aug 20, 2020 by mattjj • Draft
Simplify the interface for host_callback.id_tap
cla: yes
#4101 opened Aug 19, 2020 by shoyer
@gnecula
17
Upstream propagate interpreter from oryx to jax.experimental
cla: yes
#4088 opened Aug 17, 2020 by sharadmv
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 6 of 8
18
omnistaging on by default
cla: yes
#4038 opened Aug 12, 2020 by mattjj • Draft
3
7
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
Add GMRES to `jax.scipy.sparse.linalg`
cla: yes
#4025 opened Aug 11, 2020 by gehring • Draft 0 of 4
14
ProTip! no:milestone will show everything without a milestone.
You can’t perform that action at this time.