Skip to content

/ jax

New pull request New
137 Open 3,039 Closed
137 Open 3,039 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
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
Cleanup: remove unnecessary rng_factory boilerplate in lax_numpy_test
cla: yes pull ready
#5086 opened Dec 3, 2020 by jakevdp • Approved
lax.convert_element_type: always return DeviceArray
cla: yes
#5085 opened Dec 2, 2020 by jakevdp
[jax2tf] Added testing for the conversion of convert_element_type.
cla: yes
#5083 opened Dec 2, 2020 by bchetioui
improve sinc jvp at zero
cla: yes
#5077 opened Dec 2, 2020 by mattjj • Approved
1
5
Ensure lax.convert_element_type preserves concrete inputs
cla: yes
#5075 opened Dec 2, 2020 by jakevdp
2
[jax2tf] Add testing for the conversion of (almost) all the remaining untested primitives
cla: yes pull ready
#5066 opened Dec 1, 2020 by bchetioui • Approved
3
Implement statistic pad mode in jax.numpy.pad
cla: yes pull ready
#5060 opened Dec 1, 2020 by minoring • Approved
21
all_to_all: Add optional axis_index_groups argument
cla: yes
#5059 opened Dec 1, 2020 by jpuigcerver
7
Expose sharded_jit, PartitionSpec, and with_sharding_constraint in jax.experimental namespace.
cla: yes pull ready
#5038 opened Nov 28, 2020 by skye
2
added prepend and append to diff
cla: yes pull ready
#5035 opened Nov 28, 2020 by saklani • Approved
@jakevdp
10
GMRES cleanup/fixup
cla: yes pull ready
#5034 opened Nov 28, 2020 by shoyer
@shoyer
25
Transpose rule for imag
cla: yes pull ready
#5032 opened Nov 27, 2020 by inailuig
1
@shoyer
4
X32 tests: fail on dtype warnings
better_errors cla: yes pull ready
#5018 opened Nov 25, 2020 by jakevdp • Approved
3
Warn on host round-trip caused by jnp.array()
cla: yes
#5015 opened Nov 25, 2020 by jakevdp • Draft
[jax2tf] Add note that the TF.js conversion is experimental.
cla: yes
#4997 opened Nov 23, 2020 by bchetioui
Add dynamically methods to _DeviceArray and PyBuffer.
cla: yes pull ready
#4987 opened Nov 20, 2020 by jblespiau • Approved
4
Add vector-valued gradient example to quickstart
cla: yes
#4948 opened Nov 19, 2020 by gmittal
Improve pmap caching
cla: yes
#4923 opened Nov 17, 2020 by apaszke
Extract cloud TPU Colab set-up to a re-usable helper.
cla: yes
#4911 opened Nov 16, 2020 by aslanides
1
ci-build.yaml: mypy jax failing on the current version of Python
cla: yes
#4897 opened Nov 15, 2020 by cclauss
1
Add logging for pmap input (re)sharding
cla: yes
#4895 opened Nov 13, 2020 by jekbradbury
Cloned from CL 333372978 by 'g4 patch'.
cla: no
#4892 opened Nov 13, 2020 by copybara-service bot
1
DOC: Add description how to view pytree definition
cla: yes
#4856 opened Nov 10, 2020 by Baschdl
Fix batch polymorphism for take and gather
cla: yes pull ready
#4855 opened Nov 10, 2020 by gnecula
5
WIP: propagate weak types through unary and binary ops
cla: yes
#4850 opened Nov 10, 2020 by jakevdp • Draft
ProTip! Follow long discussions with comments:>50.
You can’t perform that action at this time.