-
Notifications
You must be signed in to change notification settings - Fork 3.2k
Pull requests: jax-ml/jax
Author
Label
Projects
Milestones
Reviews
Assignee
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] apply_vector_layout C++ rewrite (17): for.op (from cl/568376871)
#17872
by copybara-service
bot
was closed Oct 5, 2023
Loading…
random_test: Split into two so that each target is small enough to fit within a medium timeout.
#17944
by copybara-service
bot
was closed Oct 5, 2023
Loading…
[Mosaic] apply_vector_layout C++ rewrite: Have each rule use its own builder instead of passing a builder in the context
#17871
by copybara-service
bot
was closed 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…
[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…
[Mosaic] apply_vector_layout: Use shape in generalizes check (in Python)
#17942
by copybara-service
bot
was closed Oct 5, 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…
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…
[Mosaic] apply_vector_layout C++ rewrite: Handle elementwise ops by checking for the Elementwise trait and using the generic Operation interface, without templates
#17937
by copybara-service
bot
was closed Oct 9, 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…
Previous Next
ProTip!
Updated in the last three days: updated:>2025-09-30.