Skip to content

/ jax

New pull request New
140 Open 4,529 Closed
140 Open 4,529 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
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 optional fill_value argument to jnp.nonzero
cla: yes
#7592 opened Aug 11, 2021 by jakevdp
1
@jakevdp
jax.image: fix error with zero-size image resizing
cla: yes pull ready
#7591 opened Aug 11, 2021 by jakevdp • Approved
1
@jakevdp
Test failing presubmits.
cla: yes
#7588 opened Aug 11, 2021 by copybara-service bot
Change a test.
cla: no
#7587 opened Aug 11, 2021 by copybara-service bot
2
Add @jit decorators to another tranche of jax.numpy functions.
cla: yes
#7582 opened Aug 11, 2021 by copybara-service bot
Internal backend config changes.
cla: yes
#7579 opened Aug 11, 2021 by copybara-service bot
WIP: Improved x64 bookkeeping
cla: yes
#7577 opened Aug 10, 2021 by sharadmv • Draft
4
[sparse] implementing transposition rule for sparse.coo_matmat.
cla: yes
#7540 opened Aug 6, 2021 by dimasad
@jakevdp
4
Update maml.ipynb
cla: yes
#7530 opened Aug 6, 2021 by tigerneil
autodidax prototype of user-customizable partial eval strategies
cla: yes
#7517 opened Aug 5, 2021 by mattjj • Draft
@mattjj
7
Efficient C++ pmap implementation.
cla: yes
#7511 opened Aug 5, 2021 by copybara-service bot
Add post_hook support for pmap, to support debug_nans and debug_infs.
cla: yes
#7505 opened Aug 5, 2021 by copybara-service bot
Create PyBuffer lists in a ShardedDeviceArray lazilly when accessed.
cla: yes
#7504 opened Aug 5, 2021 by copybara-service bot
Use R2C FFTs to implement jnp.fft.fft() for real inputs.
cla: yes
#7494 opened Aug 4, 2021 by hawkinsp
1
3
Add jax.invertible to the JAX documentation
cla: yes
#7484 opened Aug 4, 2021 by shawwn
1
Jittable eigh
cla: yes
#7477 opened Aug 3, 2021 by adam-lewis-gcorp
@hawkinsp
1
Internal change
cla: yes
#7457 opened Aug 2, 2021 by copybara-service bot
Clean up functions from/to pybind11::tuple to std::vector<T>.
cla: yes
#7454 opened Aug 2, 2021 by copybara-service bot
Use the JIT Global & Thread-local state in PMAP.
cla: yes
#7450 opened Aug 2, 2021 by copybara-service bot
Implement squeeze masking rule
cla: yes
#7428 opened Jul 31, 2021 by shawwn
2
Initial commit of jax2iree
cla: yes
#7425 opened Jul 30, 2021 by stellaraccident • Draft
23
not going to submit
cla: yes
#7396 opened Jul 27, 2021 by copybara-service bot
In some cases axis_name is a tuple instead of string, handling it in …
cla: yes
#7393 opened Jul 27, 2021 by ahmedjawedaj
3
Replaced Any with concrete types for APIs in jaxlib.xla_client
cla: yes
#7386 opened Jul 26, 2021 by superbobry • Draft
@hawkinsp
6
Fix unused variable and warnings in docs Neural Networks Tutorial
cla: yes
#7333 opened Jul 20, 2021 by FelipeMartins96
@mattjj
ProTip! Updated in the last three days: updated:>2021-08-08.