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
Sort

Pull requests list

Fix overloaded type signature for jax.numpy.where.
#28314 opened Apr 27, 2025 by carlosgmartin Loading…
Automated Code Change
#28310 opened Apr 26, 2025 by copybara-service bot Loading…
Uses include to reference xla packages in jaxlib xla BUILD
#28304 opened Apr 25, 2025 by yliu120 Loading…
[Mosaic] Remove Python pipeline.
#28303 opened Apr 25, 2025 by copybara-service bot Loading…
Add is_leaf_with_path predicate.
#28300 opened Apr 25, 2025 by copybara-service bot Loading…
Declare tpu.vector_load, mirroring tpu.vector_store.
#28289 opened Apr 25, 2025 by copybara-service bot Loading…
Ensure __jax_array__ works properly with JIT disabled. pull ready Ready for copybara import and testing
#28288 opened Apr 25, 2025 by jakevdp Loading…
Add argnums param to fwd_and_bwd
#28279 opened Apr 25, 2025 by gspschmid Loading…
Trying to lower the degree of parallelism CI Optional GPU Presubmit Label to flag PR to run additional GPU testing not in standard presubmits
#28273 opened Apr 25, 2025 by copybara-service bot Loading…
[DO NOT MERGE] Relaxing B200 presubmit conditions
#28263 opened Apr 25, 2025 by copybara-service bot Loading…
[JAX:SPARSE] Implement CSR sparse kernel
#28261 opened Apr 25, 2025 by copybara-service bot Loading…
ProTip! Type g i on any issue or pull request to go back to the issue listing page.