The Wayback Machine - https://web.archive.org/web/20201119215809/https://github.com/google/jax/pulls
Skip to content

/ jax

New pull request New
133 Open 2,971 Closed
133 Open 2,971 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
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
jax.numpy: match numpy arguments for reductions
cla: yes pull ready
#4969 opened Nov 19, 2020 by jakevdp • Approved
[jax2tf] Added testing code for div/rem
cla: yes
#4963 opened Nov 19, 2020 by bchetioui
1
[jax2tf] Fix and test conversion of round.
cla: yes
#4959 opened Nov 19, 2020 by bchetioui
1
@gnecula
2
Add XLA lowering for xmap
cla: yes pull ready
#4954 opened Nov 19, 2020 by apaszke
Remove devices from xla.AxisEnv
cla: yes pull ready
#4953 opened Nov 19, 2020 by apaszke
Add vector-valued gradient example to quickstart
cla: yes
#4948 opened Nov 19, 2020 by gmittal
Add derivative for eigenvalues
cla: yes pull ready
#4941 opened Nov 18, 2020 by j-towns
@shoyer
11
Improve pmap caching
cla: yes
#4923 opened Nov 17, 2020 by apaszke
update jax repository citation
cla: yes
#4919 opened Nov 17, 2020 by mattjj • Approved
Extract cloud TPU Colab set-up to a re-usable helper.
cla: yes
#4911 opened Nov 16, 2020 by aslanides
1
Add support for out_axes in pmap
cla: yes pull ready
#4904 opened Nov 16, 2020 by apaszke
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
More stable implementation of np.hypot
cla: yes
#4893 opened Nov 13, 2020 by ayush-1506 • Changes requested
1
8
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
experiments in the implementation of vmap collectives
cla: yes
#4835 opened Nov 9, 2020 by mattjj • Draft 0 of 1
add gpu determinism note
cla: yes
#4824 opened Nov 7, 2020 by mattjj • Approved
4
Investigations into batch polymorphism.
cla: yes
#4817 opened Nov 6, 2020 by bchetioui • Draft 1 of 4
@gnecula
7
[RFC] Support complex dtype in random.normal
cla: yes
#4805 opened Nov 5, 2020 by PhilipVinc
9
Updates ShardingSpec's docstring
cla: yes pull ready
#4804 opened Nov 5, 2020 by marcvanzee • Approved
@gnecula
1
Internal change
cla: no
#4772 opened Nov 3, 2020 by copybara-service bot
1
add weak_type framework to abstract eval
cla: yes
#4759 opened Nov 2, 2020 by jakevdp
ProTip! Filter pull requests by the default branch with base:master.
You can’t perform that action at this time.