8000 Pull requests · jax-ml/jax · GitHub
[go: up one dir, main page]
Skip to content

Pull requests: jax-ml/jax

New pull request New
564 Open 21,999 Closed
564 Open 21,999 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
assignee:
Filter by this user
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[mutable-arrays] handle reverse in scan discharge rule
#32438 opened Oct 8, 2025 by mattjj Loading…
1
Use ifrt::LoadedExecutable::GetHumanReadableProgramText() to implement Executable.as_text().
#32435 opened Oct 8, 2025 by copybara-service bot Loading…
Use less memory during MLIR compilation.
#32428 opened Oct 7, 2025 by copybara-service bot Loading…
[JAX] Define PyUserContextScope
#32426 opened Oct 7, 2025 by copybara-service bot Loading…
[Jax] Propagate XLA metadata to function argument attributes in MLIR.
#32425 opened Oct 7, 2025 by copybara-service bot Loading…
Add pl.tpu.set_tpu_interpret_mode to globally set TPU interpret mode.
#32424 opened Oct 7, 2025 by copybara-service bot Loading…
@jburnim
Handle int4/uint4 in dtypes.to_inexact
#32423 opened Oct 7, 2025 by jakevdp Loading…
1
@jakevdp
Bump date version check.
#32419 opened Oct 7, 2025 by copybara-service bot Loading…
@mwhittaker
Update BUILD dependencies for pjrt_c_api_client.
#32416 opened Oct 7, 2025 by copybara-service bot Loading…
Update BUILD dependencies for pjrt_c_api_client.
#32415 opened Oct 7, 2025 by copybara-service bot Loading…
[Pallas:MGPU] Expose WGMMA upcast layouts to Pallas MGPU
#32407 opened Oct 7, 2025 by copybara-service bot Loading…
[Mosaic] More informative Vmem OOM error message.
#32400 opened Oct 7, 2025 by copybara-service bot Loading…
WIP: fix #32242
#32398 opened Oct 7, 2025 by sheilsarda • Draft
[Mosaic] Fix the usage of interleaved packing/unpacking to align with the semantics in tpu.td.
#32395 opened Oct 6, 2025 by copybara-service bot Loading…
Share PrimitiveTypeToNbDLDataType with xla.
#32393 opened Oct 6, 2025 by copybara-service bot Loading…
Set the default value of jax_ragged_dot_use_ragged_dot_instruction to True.
#32389 opened Oct 6, 2025 by copybara-service bot Loading…
Reverts 92f5dcba89bb833b51d4dc6cdd0b032c39e37b46
#32387 opened Oct 6, 2025 by copybara-service bot Loading…
[Mosaic TPU] Improve the layouts used in the load+reshape fusion
#32381 opened Oct 6, 2025 by copybara-service bot Loading…
@apaszke
[Mosaic GPU] Support profiling for WG semantic.
#32370 opened Oct 6, 2025 by copybara-service bot Loading…
[Mosaic GPU] Add more extensive profiler test.
#32369 opened Oct 6, 2025 by copybara-service bot Loading…
[Pallas:SC] Adds support for VMEM_SHARED and HBM scratch operands to plsc.kernel and core_map.
#32363 opened Oct 6, 2025 by copybara-service bot Loading…
Automated Code Change
#32359 opened Oct 6, 2025 by copybara-service bot Loading…
Fix: local_window_size ignored when mask is None in dot_product_attention
#32357 opened Oct 5, 2025 by akshay-babbar Loading…
1
1
Mark the pytorch interoperability test as thread-unsafe.
#32352 opened Oct 4, 2025 by copybara-service bot Loading…
@hawkinsp
[Mosaic GPU] Fix error message.
#32350 opened Oct 4, 2025 by copybara-service bot Loading…
ProTip! Type g i on any issue or pull request to go back to the issue listing page.
0