Sitelet https://web.archive.org/web/20200911123301/https://github.com/google/jax/pulls
Skip to content

/ jax

New pull request New
111 Open 2,519 Closed
111 Open 2,519 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
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
[jax2tf] Add input and output names to the TensorFlow graph generated by jax2tf.
cla: yes
#4262 opened Sep 11, 2020 by petebu
[jax2tf] Fix tests when running with JAX_ENABLE_X64=1.
cla: yes
#4261 opened Sep 11, 2020 by SIben
implement jnp.apply_along_axis
cla: yes
#4253 opened Sep 10, 2020 by jakevdp • Approved
1
[jax2tf] Added the last comments from the jax2tf doc inside the correctness_stats code.
cla: yes
#4249 opened Sep 10, 2020 by SIben
[jax2tf] Add converted primitives without tests to the generated doc.
cla: yes
#4248 opened Sep 10, 2020 by SIben
4
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
2
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
1
[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
ProTip! Adding no:label will show everything without a label.
You can’t perform that action at this time.