Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Add an
emit_python_callback helper function
#10447
opened Apr 26, 2022 by
copybara-service
bot
Loading…
Add fully asynchronous checkpointing. This will allow the training to proceed forward when the checkpoint is being committed.
#10446
opened Apr 26, 2022 by
copybara-service
bot
Loading…
[mhlo] Add result type inference for mhlo.dynamic-slice.
#10440
opened Apr 25, 2022 by
copybara-service
bot
Loading…
remove useless&expensive core.get_aval in jnp.power
#10429
opened Apr 23, 2022 by
YouJiacheng
Loading…
Add a dtypes option to cast host arrays when reloading from TS.
#10427
opened Apr 23, 2022 by
copybara-service
bot
Loading…
Annotate vmap and document pytree axis specifications
#10424
opened Apr 22, 2022 by
NeilGirdhar
Loading…
try to improve docs for scipy.linalg with unused parameters
#10419
opened Apr 22, 2022 by
YouJiacheng
Loading…
[sparse] Add BCOO attribute
_indices_sorted.
#10412
opened Apr 22, 2022 by
copybara-service
bot
Loading…
Improve performance of array integer indexing
pull ready
Ready for copybara import and testing
#10393
opened Apr 20, 2022 by
lgeiger
Loading…
[JAX] Change jnp.take_along_axis to return invalid (e.g. NaN) values for out-of-bounds indices.
#10382
opened Apr 20, 2022 by
copybara-service
bot
Loading…
[MHLO] Add MHLO lowering for erf and erfc
#10373
opened Apr 20, 2022 by
copybara-service
bot
Loading…
Change default jnp.take_along_axis gather mode to "fill".
#10356
opened Apr 19, 2022 by
copybara-service
bot
Loading…
Bump sphinx-autodoc-typehints from 1.11.1 to 1.18.1
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#10340
opened Apr 18, 2022 by
dependabot
bot
Loading…
add annotation for ndarray.at and other functions.
#10322
opened Apr 16, 2022 by
YouJiacheng
Loading…
[mesh_utils] Support creating device meshes for hybrid networks
pull ready
Ready for copybara import and testing
#10316
opened Apr 15, 2022 by
copybara-service
bot
Loading…
Delete the
mesh context manager. The replacement for it is Mesh.
#10297
opened Apr 14, 2022 by
copybara-service
bot
Loading…
Add in runtime tokens for effectful jaxprs
pull ready
Ready for copybara import and testing
#10295
opened Apr 14, 2022 by
sharadmv
Loading…
make config.jax_platforms = 'cpu' suppress no-cpu warning
#10272
opened Apr 13, 2022 by
mattjj
Loading…
Previous Next
ProTip!
Add no:assignee to see everything that’s not assigned.