Author
Label
Projects
Milestones
Reviews
Assignee
Sort
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
Recast int/bool tangents to float0 in custom_jvp/vjps (also in the
cla: yes
#4499
opened Oct 8, 2020 by
copybara-service
bot
mark lax traceables as entry points for filtered stack traces
cla: yes
#4405
opened Sep 25, 2020 by
froystig
Add a prototype implementation of recursive checkpointing
cla: yes
#4363
opened Sep 21, 2020 by
apaszke
0 of 3
[jax2tf] Added support for shape polymorphism conversion.
cla: yes
#4360
opened Sep 21, 2020 by
gnecula
Previous Next
ProTip!
Type g i on any issue or pull request to go back to the issue listing page.