-
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
Fix deactivation of array_test on certain CPU platforms.
#33236
opened Nov 11, 2025 by
copybara-service
bot
Loading…
Fix mishandling of large matrices in batched eigendecomposition.
#33233
opened Nov 11, 2025 by
copybara-service
bot
Loading…
[Mosaic] Determine matmul data format from LHS instead of RHS.
#33223
opened Nov 10, 2025 by
copybara-service
bot
Loading…
[remat] Make prevent_cse's tuple form account for consts
#33217
opened Nov 10, 2025 by
gspschmid
Loading…
Support pytree classes with dunder-style flatten/unflatten methods
#33215
opened Nov 10, 2025 by
jakevdp
Loading…
Add wheel import tests for JAX and jaxlib.
#33214
opened Nov 10, 2025 by
copybara-service
bot
Loading…
[Mosaic GPU] Add support for 8-bit types in sparse tcgen05 MMA
#33213
opened Nov 10, 2025 by
copybara-service
bot
Loading…
[Mosaic GPU] Add support for integer WGMMA with LHS in registers under WG semantic.
#33212
opened Nov 10, 2025 by
copybara-service
bot
Loading…
[Mosaic GPU] Grow k_group_elems instead of complaining about small swizzle
#33209
opened Nov 10, 2025 by
copybara-service
bot
Loading…
[Mosaic GPU] Support nvfp4 scaled MMA instructions
#33208
opened Nov 10, 2025 by
copybara-service
bot
Loading…
Add support for global scale to
scaled_matmul_wrapper
#33204
opened Nov 10, 2025 by
sergey-kozub
Loading…
[Mosaic GPU] Add custom vector load/store ops to the dialect.
#33202
opened Nov 10, 2025 by
copybara-service
bot
Loading…
Fix ROCm lib names copy pattern to use wildcard version
#33187
opened Nov 7, 2025 by
phambinhfin
Loading…
Skip
test_oom_compilation_only on TPU v6e and v7x
#33185
opened Nov 7, 2025 by
copybara-service
bot
Loading…
[Mosaic GPU] Add support for block-scaled MMA with all K sizes divisible by 128
#33173
opened Nov 7, 2025 by
copybara-service
bot
Loading…
Add a test for a compiler bug related to int div/mod.
#33170
opened Nov 7, 2025 by
copybara-service
bot
Loading…
[Mosaic TPU][NFC] Share checks between the reshape/store and load/reshape optimization rules
#33163
opened Nov 7, 2025 by
copybara-service
bot
Loading…
Fix Python path for executing "hi" JAX code.
#33161
opened Nov 7, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
no:milestone will show everything without a milestone.