Skip to content

Pull requests: google/jax

New pull request New
376 Open 12,891 Closed
376 Open 12,891 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 Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Try to fix a compilation_cache_test testcase in CI.
#20964 opened Apr 26, 2024 by copybara-service bot Loading…
1
Added elementwise_inline_asm to Pallas GPU
#20962 opened Apr 26, 2024 by copybara-service bot Loading…
@superbobry
DOC: link docstrings to github source
documentation
#20961 opened Apr 26, 2024 by jakevdp Loading…
@jakevdp
Update Tensorstore dtype spec building to support int4
#20960 opened Apr 26, 2024 by copybara-service bot Loading…
DOC: improve docs of transpose & matrix_transpose
documentation pull ready Ready for copybara import and testing
#20957 opened Apr 26, 2024 by jakevdp Loading…
@jakevdp
Fix imports in Mosaic GPU examples
#20953 opened Apr 26, 2024 by copybara-service bot Loading…
@apaszke
Add get_jaxpr on jax.jit so that we can delete the duplicate code in jax.make_jaxpr.
#20942 opened Apr 25, 2024 by copybara-service bot Loading…
@yashk2810
DOC: improve documentation of where, nonzero, and related functions
documentation pull ready Ready for copybara import and testing
#20941 opened Apr 25, 2024 by jakevdp Loading…
@jakevdp
Implements Ragged Dot API
#20940 opened Apr 25, 2024 by piotrfilipiuk Loading…
@mattjj
6
Enable async dispatch on JAX CPU by setting 'jax_cpu_enable_async_dispatch' to be True by default.
#20939 opened Apr 25, 2024 by copybara-service bot Loading…
@yueshengys
Remove extraneous pure_callback_api wrapper
#20937 opened Apr 25, 2024 by jakevdp Loading…
1
@jakevdp
API deprecation doc: explicitly list some dates
documentation
#20936 opened Apr 25, 2024 by jakevdp Loading…
@jakevdp
Ported LuPivotsToPermutation to the typed XLA FFI
#20931 opened Apr 25, 2024 by copybara-service bot Loading…
@superbobry
Workaround mpmath 1.3 issues in asinh evaluation at infinities
#20923 opened Apr 24, 2024 by pearu Loading…
Remove type promotion for mixed Nvidia's fp8 matmuls.
#20906 opened Apr 23, 2024 by elfiegg Loading…
@hawkinsp
Make Tracers non-hashable
kokoro:force-run pull ready Ready for copybara import and testing
#20903 opened Apr 23, 2024 by jakevdp Loading…
@jakevdp
Propagate the available axes to the partitioning function
#20884 opened Apr 23, 2024 by superbobry Loading…
1
[ROCm] Add float8_e4m3fnuz test
#20880 opened Apr 23, 2024 by Ruturaj4 Loading…
3
Add get_device_ordinal_from_cuda_array_interface to cuda plugin so that CUDA dependency can be removed from py_array (jaxlib).
#20879 opened Apr 23, 2024 by copybara-service bot Loading…
@jyingl3
In QDWH, eliminate one of the two triangular solves in the steps using Cholesky.
#20878 opened Apr 23, 2024 by copybara-service bot Loading…
DOC: Remove dependency on sphinx_autodoc_typehints
documentation pull ready Ready for copybara import and testing
#20875 opened Apr 22, 2024 by jakevdp Loading…
@jakevdp
[Mosaic] Add support for remote DMAs and semaphores in megacore mode
#20860 opened Apr 22, 2024 by copybara-service bot Loading…
@apaszke
jax.scipy.stats: manually document functions to avoid scipy import
documentation
#20847 opened Apr 19, 2024 by jakevdp Loading…
@jakevdp
18
Internal.
#20845 opened Apr 19, 2024 by copybara-service bot Loading…
Refactored common upcast for integral-type accumulators
#20842 opened Apr 19, 2024 by Micky774 Loading…
@jakevdp
4
ProTip! Mix and match filters to narrow down what you’re looking for.