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

Bump shard_count for shard_map_test to fix timeouts.
#17956 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…
Expose mlir.ShapePolyLoweringState
#17955 by copybara-service bot was closed Oct 5, 2023 Loading…
Re-add JIT to jax.numpy.bitwise_count
#17963 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…
[random] wrap_key_data accepts impl=key.dtype.impl pull ready Ready for copybara import and testing
#17768 by jakevdp was closed Oct 6, 2023 Loading…
fix typo in scatter_add doc
#18014 by andrinr was closed Oct 9, 2023 Loading…
Migrate a subset of internal modules to use state objects
#18008 by copybara-service bot was closed Oct 9, 2023 Loading…
Increase minimum scipy version to 1.9.
#18016 by copybara-service bot was closed Oct 9, 2023 Loading…
jnp.frompyfunc: fix .at() edge case pull ready Ready for copybara import and testing
#18017 by jakevdp was merged Oct 9, 2023 Loading…
Only perform compilation cache writes from process 0.
#17967 by copybara-service bot was closed Oct 9, 2023 Loading…
Remove take_ownership support in DLPack.
#17941 by copybara-service bot was closed Oct 10, 2023 Loading…
[pallas] Simplify several primitives bindings.
#18043 by copybara-service bot was closed Oct 10, 2023 Loading…
ProTip! Adding no:label will show everything without a label.