Skip to content

/ jax

New pull request New
118 Open 2,694 Closed
118 Open 2,694 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 extraneous device_put
cla: yes
#4530 opened Oct 9, 2020 by jakevdp
Enable the C++ jax.jit fast code-path by default.
cla: yes
#4529 opened Oct 9, 2020 by copybara-service bot
Add support for jax_debug_nans and fix the last few glitches with the C++ jax.jit.
cla: yes
#4527 opened Oct 9, 2020 by copybara-service bot
Enable the C++ jax.jit fast code-path by default.
cla: yes
#4525 opened Oct 9, 2020 by copybara-service bot
[jax2tf] Add paths that do not use XLA in conv_general_dilated.
cla: yes
#4522 opened Oct 9, 2020 by bchetioui
pretty print improvements
cla: yes
#4513 opened Oct 8, 2020 by mattjj • Draft
Mark pmin and pmax as multi-host supported.
cla: yes pull ready
#4509 opened Oct 8, 2020 by malmaud • Approved
1
Fallback when faced with other runtimes.
cla: yes
#4506 opened Oct 8, 2020 by copybara-service bot
Recast int/bool tangents to float0 in custom_jvp/vjps (also in the
cla: yes
#4499 opened Oct 8, 2020 by copybara-service bot
1
Copybara import of the project:
cla: yes
#4469 opened Oct 7, 2020 by copybara-service bot
Add `lax.conv_general_dilated_patches`
cla: yes
#4459 opened Oct 6, 2020 by romanngg
8
Histogram2d implementation
cla: yes pull ready
#4451 opened Oct 4, 2020 by alexminnaar • Approved
12
wrap long line
cla: yes pull ready
#4440 opened Oct 2, 2020 by froystig
@froystig
Add support for creating float0 arrays.
cla: yes
#4437 opened Oct 2, 2020 by tomhennigan
1
6
WIP: add experimental sparse support
cla: yes
#4422 opened Sep 29, 2020 by jakevdp • Draft
mark lax traceables as entry points for filtered stack traces
cla: yes
#4405 opened Sep 25, 2020 by froystig
@froystig
2
avals with names
cla: yes
#4397 opened Sep 24, 2020 by jekbradbury • Draft
1
[jax2tf][js] Adding test utils to analyze TF.js conversion possibilities (WIP)
cla: yes
#4394 opened Sep 23, 2020 by bchetioui • Draft
[jax2tf] Implement proper tracking of which primitives have dedicated tests (WIP)
cla: yes
#4375 opened Sep 22, 2020 by bchetioui • Draft
1
Add a prototype implementation of recursive checkpointing
cla: yes
#4363 opened Sep 21, 2020 by apaszke 0 of 3
13
[jax2tf] Added support for shape polymorphism conversion.
cla: yes
#4360 opened Sep 21, 2020 by gnecula
2
Added all to all batching support
cla: yes
#4357 opened Sep 20, 2020 by Thenerdstation • Draft
1
Implicitly restarted arnoldi
cla: yes
#4336 opened Sep 18, 2020 by mganahl • Draft
4
[jax2tf] Add support for Add for uint32
cla: yes
#4334 opened Sep 18, 2020 by gnecula
1
Solving #2347
cla: yes
#4323 opened Sep 17, 2020 by tudorcebere • Approved
13
ProTip! Type g i on any issue or pull request to go back to the issue listing page.
You can’t perform that action at this time.