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

/ jax

New pull request New
128 Open 2,880 Closed
128 Open 2,880 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.scipy.signal functions for n-dim arrays
cla: yes
#4822 opened Nov 7, 2020 by ayush-1506
1
2
PRNGKey: more careful handling of typecasting for various inputs.
cla: yes
#4821 opened Nov 6, 2020 by jakevdp
DOC: fix generating code in type_promotion.rst
cla: yes
#4820 opened Nov 6, 2020 by jakevdp
Investigations into batch polymorphism.
cla: yes
#4817 opened Nov 6, 2020 by bchetioui • Draft 1 of 4
5
[jax2tf] Add the convert_and_save_model to the jax2tf API.
cla: yes pull ready
#4816 opened Nov 6, 2020 by gnecula
3
[DOCS] [cosmetic] Add JAX favicon to sphinx docs
cla: yes
#4815 opened Nov 6, 2020 by j-towns
2
Make ShardingSpec more flexible
cla: yes pull ready
#4814 opened Nov 6, 2020 by apaszke
[DOCS] Improve docs for custom_jvp and custom_vjp
cla: yes
#4812 opened Nov 6, 2020 by j-towns
[RFC] Support complex dtype in random.normal
cla: yes
#4805 opened Nov 5, 2020 by PhilipVinc
Updates ShardingSpec's docstring
cla: yes
#4804 opened Nov 5, 2020 by marcvanzee
Fix doc for index operations
cla: yes
#4803 opened Nov 5, 2020 by AdrienCorenflos
[jax2tf] Adding lax.named_call_p to the list of unexpected primitives.
cla: yes pull ready
#4801 opened Nov 5, 2020 by bchetioui
Add a named_call public function in lax
cla: yes
#4796 opened Nov 5, 2020 by qiuminxu
9
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
WIP pod sharded_jit prototype
cla: yes pull ready
#4746 opened Oct 31, 2020 by skye • Draft
an experiment in handling instances with __array__
cla: yes
#4725 opened Oct 28, 2020 by mattjj • Draft 0 of 4
5
Add typing annotations and expose py.typed
cla: yes
#4711 opened Oct 26, 2020 by NeilGirdhar
make the custom_vjp bwd None handling more robust
cla: yes
#4683 opened Oct 22, 2020 by mattjj
1
[x64] Add initial weak_type logic to abstract eval rules
cla: yes
#4682 opened Oct 22, 2020 by jakevdp
1
implicitly restarted lanczos method
cla: yes
#4675 opened Oct 22, 2020 by mganahl
2
Add `lax.conv_general_dilated_local`
cla: yes
#4671 opened Oct 21, 2020 by romanngg
Internal change
cla: no
#4669 opened Oct 21, 2020 by copybara-service bot
5
Preserve name scope of named_call primitive in jax2tf
cla: yes
#4649 opened Oct 20, 2020 by qiuminxu • Approved
@gnecula
15
Add fast path for jnp.repeat when repeats is a scalar constant.
cla: yes pull ready
#4639 opened Oct 19, 2020 by petebu • Approved
7
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.
You can’t perform that action at this time.