Skip to content

Pull requests: jax-ml/jax

Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort

Pull requests list

Simplify the Bazel logic used to add dependencies to tests. pull ready Ready for copybara import and testing
#28753 by hawkinsp was merged May 14, 2025 Loading…
jax-cuda12-plugin: require nvidia-cublas-cu12<12.9 kokoro:force-run pull ready Ready for copybara import and testing
#28513 by olupton was merged May 5, 2025 Loading…
Reenable CUDA version checks from Python. kokoro:force-run pull ready Ready for copybara import and testing
#28755 by hawkinsp was merged May 15, 2025 Loading…
[run_state] add scan nested test, tweak rule name to mention 'state' pull ready Ready for copybara import and testing
#17928 by mattjj was merged Oct 6, 2023 Loading…
Bump shard_count for shard_map_test to fix timeouts.
#17956 by copybara-service bot was closed Oct 5, 2023 Loading…
Asynchronous dispatch doc update regarding jax.Array migration pull ready Ready for copybara import and testing
#17965 by tttttanya was merged Oct 5, 2023 Loading…
[Mosaic] infer_memref_layout C++ rewrite
#17940 by copybara-service bot was closed Oct 5, 2023 Loading…
Update setup.py and CHANGELOG for jax 0.4.18 release pull ready Ready for copybara import and testing
#17989 by skye was merged Oct 6, 2023 Loading…
Expose mlir.ShapePolyLoweringState
#17955 by copybara-service bot was closed Oct 5, 2023 Loading…
[Mosaic] apply_vector_layout C++ rewrite: various bug fixes
#17939 by copybara-service bot was closed Oct 5, 2023 Loading…
Update versions post jax 0.4.18 release kokoro:force-run pull ready Ready for copybara import and testing
#17994 by skye was merged Oct 7, 2023 Loading…
fix pow jvp rule with int exponent (broken since #16419) kokoro:force-run pull ready Ready for copybara import and testing
#17996 by mattjj was merged Oct 7, 2023 Loading…
Improve CUDA install documenation. kokoro:force-run pull ready Ready for copybara import and testing
#17981 by hawkinsp was merged Oct 6, 2023 Loading…
Fix tests that mistakenly used assertRaises(..., msg=...) to match… pull ready Ready for copybara import and testing
#17979 by hawkinsp was merged Oct 6, 2023 Loading…
fix typo in scatter_add doc pull ready Ready for copybara import and testing
#18015 by andrinr was merged Oct 9, 2023 Loading…
fix typo in scatter_add doc
#18014 by andrinr was closed Oct 9, 2023 Loading…
Ignore patch versions of cufft and cusolver in CUDA version checks. kokoro:force-run pull ready Ready for copybara import and testing
#18012 by hawkinsp was merged Oct 9, 2023 Loading…
Fix NumPy warning when casting negative float to unsigned in tan() te… pull ready Ready for copybara import and testing
#17971 by hawkinsp was merged Oct 6, 2023 Loading…
Check dtypes in fft_p's abstract eval rule. pull ready Ready for copybara import and testing
#17978 by hawkinsp was merged Oct 6, 2023 Loading…
Update README.md so that CPU install instruction works pull ready Ready for copybara import and testing
#18035 by bb515 was merged Oct 10, 2023 Loading…
ProTip! Updated in the last three days: updated:>2025-09-30.