Pull requests: jax-ml/jax
Pull requests list
[pallas:mosaic_gpu] Added support for unrolling to
lax.fori_loop lowering#28708 openedMay 13, 2025 by copybara-service bot Loading…
Use DmaCopyChunk::Make because directly assigning the struct
#28703 openedMay 13, 2025 by copybara-service bot Loading…
Infer sharding from arguments only if all mesh axes are explicit. Error out requiring in_specs/in_axes in all other cases.
#28702 openedMay 13, 2025 by copybara-service bot Loading…
Add use_raw_buffers which allows switching the implementation to
#28701 openedMay 12, 2025 by copybara-service bot Loading…
[pallas] Pulled
runtime_assert_enabled from pltpu to pl#28699 openedMay 12, 2025 by copybara-service bot Loading…
Bump hypothesis from 6.102.4 to 6.131.15 dependenciesPull requests that update a dependency file pythonPull requests that update Python code
#28693 openedMay 12, 2025 by dependabot bot Loading…
Deprecate the no-op custom_jvp_call_jaxpr_p import stub.
#28691 openedMay 12, 2025 by copybara-service bot Loading…
Revert pytest: use importlib mode by default kokoro:force-run pull readyReady for copybara import and testing
#28690 openedMay 12, 2025 by jakevdp Loading…
[CI] Add additional hardware to continuous non-rbe testing
#28688 openedMay 12, 2025 by copybara-service bot Loading…
[shard-map] start adding systematic smap tests kokoro:force-run pull readyReady for copybara import and testing
#28665 openedMay 10, 2025 by mattjj Loading…
Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with beta==0) kokoro:force-run pull readyReady for copybara import and testing
#28635 openedMay 9, 2025 by psanal35 Loading…
Ensure __jax_array__ support in binary ops kokoro:force-run pull readyReady for copybara import and testing
#28630 openedMay 8, 2025 by jakevdp Loading…
ProTip! Exclude everything labeled
bug with -label:bug.