Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
469 Open 15,022 Closed
469 Open 15,022 Closed
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
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

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…
😕 1
Bump shard_count for shard_map_test to fix timeouts.
#17956 by copybara-service bot was closed Oct 5, 2023 Loading…
1
[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…
1
[JAX] Keep CPU host callbacks alive via IFRT, rather than by attaching them to the Python object.
#17946 by copybara-service bot was closed Oct 5, 2023 Loading…
@hawkinsp
[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…
[export] Fix dot_general multi-platform lowering
#17951 by copybara-service bot was closed Oct 6, 2023 Loading…
@gnecula
Expose mlir.ShapePolyLoweringState
#17955 by copybara-service bot was closed Oct 5, 2023 Loading…
1
Re-add JIT to jax.numpy.bitwise_count
#17963 by copybara-service bot was closed Oct 5, 2023 Loading…
@jakevdp
[Mosaic] apply_vector_layout C++ rewrite: various bug fixes
#17939 by copybara-service bot was closed Oct 5, 2023 Loading…
[random] wrap_key_data accepts impl=key.dtype.impl pull ready Ready for copybara import and testing
#17768 by