Pull requests: jax-ml/jax
Pull requests list
Use DmaCopyChunk::Make because directly assigning the struct
#28703
opened May 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
opened May 13, 2025 by
copybara-service
bot
Loading…
Add use_raw_buffers which allows switching the implementation to
#28701
opened May 12, 2025 by
copybara-service
bot
Loading…
[pallas] Pulled
runtime_assert_enabled from pltpu to pl
#28699
opened May 12, 2025 by
copybara-service
bot
Loading…
Update tridiagonal solve kernels on GPU to properly use the FFI.
#28696
opened May 12, 2025 by
copybara-service
bot
Loading…
Bump hypothesis from 6.102.4 to 6.131.15
dependencies
Pull requests that update a dependency file
python
Pull requests that update Python code
#28693
opened May 12, 2025 by
dependabot
bot
Loading…
Deprecate the no-op custom_jvp_call_jaxpr_p import stub.
#28691
opened May 12, 2025 by
copybara-service
bot
Loading…
Revert pytest: use importlib mode by default
kokoro:force-run
pull ready
Ready for copybara import and testing
#28690
opened May 12, 2025 by
jakevdp
Loading…
[CI] Add additional hardware to continuous non-rbe testing
#28688
opened May 12, 2025 by
copybara-service
bot
Loading…
[shard-map] start adding systematic smap tests
kokoro:force-run
pull ready
Ready for copybara import and testing
#28665
opened May 10, 2025 by
mattjj
Loading…
Fixed @@pypi//nvidia_nvshmem_cu12 errors.
#28658
opened May 9, 2025 by
copybara-service
bot
Loading…
Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with beta==0)
kokoro:force-run
pull ready
Ready for copybara import and testing
#28635
opened May 9, 2025 by
psanal35
Loading…
Ensure __jax_array__ support in binary ops
kokoro:force-run
pull ready
Ready for copybara import and testing
#28630
opened May 8, 2025 by
jakevdp
Loading…
ProTip!
Find all pull requests that aren't related to any open issues with -linked:issue.