Skip to content

/ jax

New pull request New
139 Open 4,555 Closed
139 Open 4,555 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
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
user-customizable gradient checkpointing strategies
cla: yes pull ready
#7631 opened Aug 15, 2021 by mattjj
1 of 2 tasks
@mattjj
Don't initialize all backends when get_backend() is called with a specific backend.
cla: yes pull ready
#7628 opened Aug 14, 2021 by skye • Approved
@skye
2
Add xla_bridge_test.GetBackendTest
cla: yes pull ready
#7627 opened Aug 14, 2021 by skye • Approved
@skye
2
Add basic support for scipy.fft.dct/dctn
cla: yes
#7617 opened Aug 13, 2021 by juliuskunze • Changes requested
@jakevdp
7
mypy export of jax.numpy namespace
cla: yes
#7606 opened Aug 12, 2021 by jakevdp
1
@jakevdp
Add warning when using persistent compilation cache on non-TPU platform.
cla: yes pull ready
#7604 opened Aug 12, 2021 by skye
Drop out-of-bounds indices in gather.
cla: yes
#7595 opened Aug 11, 2021 by hawkinsp • Draft
@hawkinsp
Store avals in literals to avoid recanonicalization
cla: yes
#7577 opened Aug 10, 2021 by sharadmv
7
[sparse] implementing transposition rule for sparse.coo_matmat.
cla: yes
#7540 opened Aug 6, 2021 by dimasad
@jakevdp
4
Update maml.ipynb
cla: yes pull ready
#7530 opened Aug 6, 2021 by tigerneil
@gnecula
1
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
@hawkinsp
1
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
@gnecula
2
Initial commit of jax2iree
cla: yes
#7425 opened Jul 30, 2021 by stellaraccident • Draft
24
not going to submit
cla: yes
#7396 opened Jul 27, 2021 by copybara-service bot
@zhangqiaorjc
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 • Approved
@hawkinsp
6
Fix unused variable and warnings in docs Neural Networks Tutorial
cla: yes
#7333 opened Jul 20, 2021 by FelipeMartins96
@mattjj
Don't wrap static arguments in hashable wrappers in pmap.
cla: yes pull ready
#7328 opened Jul 19, 2021 by hawkinsp • Approved
@hawkinsp
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.