Skip to content
New pull request New
187 Open 6,105 Closed
187 Open 6,105 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 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…
@yashk2810
[mhlo] Add result type inference for mhlo.dynamic-slice.
#10440 opened Apr 25, 2022 by copybara-service bot Loading…
@hawkinsp
remove useless&expensive core.get_aval in jnp.power
#10429 opened Apr 23, 2022 by YouJiacheng Loading…
@hawkinsp
3
Add a dtypes option to cast host arrays when reloading from TS.
#10427 opened Apr 23, 2022 by copybara-service bot Loading…
@yashk2810
Add fast path for jnp.diagonal(a, offset=0)
#10426 opened Apr 22, 2022 by lgeiger Loading…
@hawkinsp
3
Annotate vmap and document pytree axis specifications
#10424 opened Apr 22, 2022 by NeilGirdhar Loading…
@jakevdp
11
try to improve docs for scipy.linalg with unused parameters
#10419 opened Apr 22, 2022 by YouJiacheng Loading…
@hawkinsp
3
[sparse] Add BCOO attribute _indices_sorted.
#10412 opened Apr 22, 2022 by copybara-service bot Loading…
@jakevdp
replace int with operator.index for lax
#10398 opened Apr 21, 2022 by YouJiacheng Loading…
21
prefer operator.index over int
#10395 opened Apr 21, 2022 by YouJiacheng • Draft
13
Improve performance of array integer indexing
pull ready
#10393 opened Apr 20, 2022 by lgeiger Loading…
@jakevdp
37
[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…
@hawkinsp
[MHLO] Add MHLO lowering for erf and erfc
#10373 opened Apr 20, 2022 by copybara-service bot Loading…
@hawkinsp
Change default jnp.take_along_axis gather mode to "fill".
#10356 opened Apr 19, 2022 by copybara-service bot Loading…
@hawkinsp
[linalg] debug detrend.
#10343 opened Apr 18, 2022 by copybara-service bot Loading…
Bump sphinx-autodoc-typehints from 1.11.1 to 1.18.1
dependencies python
#10340 opened Apr 18, 2022 by dependabot bot Loading…
1
add annotation for ndarray.at and other functions.
#10322 opened Apr 16, 2022 by YouJiacheng Loading…
28
Attempt to add Generic Wrapped
#10321 opened Apr 16, 2022 by YouJiacheng Loading…
@froystig
4
[mesh_utils] Support creating device meshes for hybrid networks
pull ready
#10316 opened Apr 15, 2022 by copybara-service bot Loading…
@jekbradbury
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
#10295 opened Apr 14, 2022 by sharadmv Loading…
@sharadmv
Simplify util.py
#10285 opened Apr 14, 2022 by YouJiacheng Loading…
@jakevdp
4
Make safe_zip and safe_map work with empty lists.
#10274 opened Apr 14, 2022 by romanngg Loading…
@mattjj
12
make config.jax_platforms = 'cpu' suppress no-cpu warning
#10272 opened Apr 13, 2022 by mattjj Loading…
4
ProTip! Add no:assignee to see everything that’s not assigned.